added tests to test stream

davinirjr · Sep 22, 2016 · 8df6c86 · 8df6c86
1 parent 8f308c4
commit 8df6c86
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_stream/test_stream.py b/tests/test_stream/test_stream.py
@@ -15,12 +15,12 @@ def test_run(self):
         outs = ConnectionSet([output])
         input.entities.extend([1, 2, 3, 4, 5, 6, 7, 8, 9, 0])
 
-        stream.run(ins, outs)
+        self.assertTrue(stream.run(ins, outs))
         self.assertEqual([2, 4, 6, 8, 10], list(output.entities))
-        stream.run(ins, outs)
+        self.assertTrue(stream.run(ins, outs))
         self.assertEqual([2, 4, 6, 8, 10, 12], list(output.entities))
         output.entities.clear()
-        stream.run(ins, outs)
+        self.assertFalse(stream.run(ins, outs))
         self.assertEqual([14, 16, 18, 0], list(output.entities))