summaryrefslogtreecommitdiff
path: root/test/test_package.py
blob: aa42c9cecfcdc7d2d98c6a03b5a43a2cf5ed1996 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class TestPackage:
    def test_top_level_namespace(self):
        import kafka as kafka1
        assert kafka1.KafkaConsumer.__name__ == "KafkaConsumer"
        assert kafka1.consumer.__name__ == "kafka.consumer"
        assert kafka1.codec.__name__ == "kafka.codec"

    def test_submodule_namespace(self):
        import kafka.client_async as client1
        assert client1.__name__ == "kafka.client_async"

        from kafka import client_async as client2
        assert client2.__name__ == "kafka.client_async"

        from kafka.client_async import KafkaClient as KafkaClient1
        assert KafkaClient1.__name__ == "KafkaClient"

        from kafka import KafkaClient as KafkaClient2
        assert KafkaClient2.__name__ == "KafkaClient"

        from kafka.codec import gzip_encode as gzip_encode1
        assert gzip_encode1.__name__ == "gzip_encode"

        from kafka.codec import snappy_encode
        assert snappy_encode.__name__ == "snappy_encode"