1 files changed, 29 insertions, 0 deletions
diff --git a/test/unit.py b/test/unit.py
index 6ca3564..7cb1aed 100644
--- a/test/unit.py
+++ b/test/unit.py
@@ -12,6 +12,35 @@ STRLEN = 100
 def random_string():
     return os.urandom(random.randint(0, STRLEN))
 
+class TestPackage(unittest.TestCase):
+    def test_top_level_namespace(self):
+        import kafka as kafka1
+        self.assertEquals(kafka1.KafkaClient.__name__, "KafkaClient")
+        self.assertEquals(kafka1.gzip_encode.__name__, "gzip_encode")
+        self.assertEquals(kafka1.client.__name__, "kafka.client")
+        self.assertEquals(kafka1.codec.__name__, "kafka.codec")
+
+    def test_submodule_namespace(self):
+        import kafka.client as client1
+        self.assertEquals(client1.__name__, "kafka.client")
+        self.assertEquals(client1.KafkaClient.__name__, "KafkaClient")
+
+        from kafka import client as client2
+        self.assertEquals(client2.__name__, "kafka.client")
+        self.assertEquals(client2.KafkaClient.__name__, "KafkaClient")
+
+        from kafka.client import KafkaClient as KafkaClient1
+        self.assertEquals(KafkaClient1.__name__, "KafkaClient")
+
+        from kafka.codec import gzip_encode as gzip_encode1
+        self.assertEquals(gzip_encode1.__name__, "gzip_encode")
+
+        from kafka import KafkaClient as KafkaClient2
+        self.assertEquals(KafkaClient2.__name__, "KafkaClient")
+
+        from kafka import gzip_encode as gzip_encode2
+        self.assertEquals(gzip_encode2.__name__, "gzip_encode")
+
 class TestMisc(unittest.TestCase):
     def test_length_prefix(self):
         for i in xrange(ITERATIONS):