def test_consume(self):
stream = Stream()
# test consume without filter
stream.queue.put(1)
stream.queue.put(2)
self.assertEqual(stream.consume(), 1)
self.assertEqual(stream.consume(), 2)
# test consume with filter
stream.queue.put(1)
stream.queue.put(2)
stream.queue.put(3)
self.assertEqual(stream.consume(lambda data: data == 2), 2)
self.assertEqual(stream.consume(), 3)
