def test_produce(self): test = [] def test_produce_cb(stream, data): test.append(data) stream = Stream() stream.produce_cb = test_produce_cb stream.produce("test") self.assertEqual(test, ["test"])