def test_flumine_stream_on_process(self): mock_listener = mock.Mock() stream = datastream.FlumineStream(mock_listener, 0) stream.on_process([1, 2, 3]) mock_listener.output_queue.put.called_with(datastream.RawDataEvent([1, 2, 3]))
def test_flumine_stream(self): mock_listener = mock.Mock() stream = datastream.FlumineStream(mock_listener, 0) self.assertEqual(str(stream), "FlumineStream") self.assertEqual(repr(stream), "<FlumineStream [0]>")