fix import error

This commit is contained in:
Muzahid 2021-06-22 18:37:46 +05:30
parent f01dfb54c9
commit c61cc1dd1f
Signed by untrusted user: cis-muzahid
GPG Key ID: 1DC85E7D3AB613EA

View File

@ -14,7 +14,10 @@ class TestNetworkGroup(unittest.TestCase):
"""
def test_network_group(self):
"""Test various types of network groups"""
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))