

def test_collect_hosts__happy_path(self):
    hosts = '127.0.0.1:1234,127.0.0.1'
    results = collect_hosts(hosts)
    self.assertEqual(set(results), set([('127.0.0.1', 1234, socket.AF_INET), ('127.0.0.1', 9092, socket.AF_INET)]))

dataset/ETHPy150Open dpkp/kafka-python/test/test_conn_legacy.py ConnTest.test_collect_hosts__happy_path/original