From c61cc1dd1f34bd9b5eece08229a61e6d7f061576 Mon Sep 17 00:00:00 2001 From: Muzahid Date: Tue, 22 Jun 2021 18:37:46 +0530 Subject: [PATCH] fix import error --- src/tests/test_networkgroup.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/tests/test_networkgroup.py b/src/tests/test_networkgroup.py index d0360276..f9d8e02d 100644 --- a/src/tests/test_networkgroup.py +++ b/src/tests/test_networkgroup.py @@ -14,7 +14,10 @@ class TestNetworkGroup(unittest.TestCase): """ def test_network_group(self): """Test various types of network groups""" - from pybitmessage.protocol import network_group + try: + from pybitmessage.protocol import network_group + except ModuleNotFoundError: + from .protocol import network_group test_ip = '1.2.3.4' self.assertEqual('\x01\x02', network_group(test_ip))