Remove suffix from transports_nb

2 files changed, 6 insertions, 6 deletions

diff --git a/test/unit/test_xmpp_transports_nb.py b/test/unit/test_xmpp_transports.py
index 0f60737..e79854b 100644
--- a/test/unit/test_xmpp_transports_nb.py
+++ b/test/unit/test_xmpp_transports.py
@@ -4,7 +4,7 @@ Unit test for tranports classes.
 
 import unittest
 
-from nbxmpp import transports_nb
+from nbxmpp import transports
 
 
 class TestModuleLevelFunctions(unittest.TestCase):
@@ -13,7 +13,7 @@ class TestModuleLevelFunctions(unittest.TestCase):
     '''
     def test_urisplit(self):
         def check_uri(uri, proto, host, port, path):
-            _proto, _host, _port, _path = transports_nb.urisplit(uri)
+            _proto, _host, _port, _path = transports.urisplit(uri)
             self.assertEqual(proto, _proto)
             self.assertEqual(host, _host)
             self.assertEqual(path, _path)
@@ -30,7 +30,7 @@ class TestModuleLevelFunctions(unittest.TestCase):
 
     def test_get_proxy_data_from_dict(self):
         def check_dict(proxy_dict, host, port, user, passwd):
-            _host, _port, _user, _passwd = transports_nb.get_proxy_data_from_dict(
+            _host, _port, _user, _passwd = transports.get_proxy_data_from_dict(
                     proxy_dict)
             self.assertEqual(_host, host)
             self.assertEqual(_port, port)
diff --git a/test/unit/test_xmpp_transports_nb2.py b/test/unit/test_xmpp_transports2.py
index 7caeda8..0bdd721 100644
--- a/test/unit/test_xmpp_transports_nb2.py
+++ b/test/unit/test_xmpp_transports2.py
@@ -7,7 +7,7 @@ import unittest
 import socket
 
 from test.lib.xmpp_mocks import IdleQueueThread, IdleMock
-from nbxmpp import transports_nb
+from nbxmpp import transports
 
 
 class AbstractTransportTest(unittest.TestCase):
@@ -72,7 +72,7 @@ class TestNonBlockingTCP(AbstractTransportTest):
             except socket.error as e:
                 self.testcase.fail(msg=str(e))
 
-            self.socket = transports_nb.NonBlockingTCP(
+            self.socket = transports.NonBlockingTCP(
                     raise_event=lambda event_type, data: self.testcase.assertTrue(
                             event_type and data),
                     on_disconnect=lambda: self.on_success(mode='SocketDisconnect'),
@@ -210,7 +210,7 @@ class TestNonBlockingHTTP(AbstractTransportTest):
     }
 
     def _get_transport(self, http_dict, proxy_dict=None):
-        return transports_nb.NonBlockingHTTP(
+        return transports.NonBlockingHTTP(
                 raise_event=None,
                 on_disconnect=None,
                 idlequeue=self.idlequeue_thread.iq,