Refactoring Structure

alemazzo · alemazzo · commit 8033ac4cd4bc · 2021-03-06T11:52:27.000+01:00
diff --git a/.gitignore b/.gitignore
@@ -2,5 +2,4 @@ __pycache__/
 build/
 dist/
 java_stream.egg-info/
-main.py
-performance.py
+main.py
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -0,0 +1,33 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python: Current File",
+            "type": "python",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal"
+        },
+        {
+            "name": "Tests: Stream",
+            "type": "python",
+            "request": "launch",
+            "module": "tests.stream"
+        },
+        {
+            "name": "Tests: Boolean",
+            "type": "python",
+            "request": "launch",
+            "module": "tests.boolean"
+        },
+        {
+            "name": "All Tests",
+            "type": "python",
+            "request": "launch",
+            "module": "tests"
+        }
+    ]
+}
diff --git a/tests/boolean/test.py b/tests/boolean/test.py
@@ -4,6 +4,11 @@
 
 class TestBooleanStream(unittest.TestCase):
 
+    def test_random(self):
+        s = BooleanStream.random().limit(20)
+        for elem in s:
+            self.assertIn(elem, [True, False])
+
     def test_true(self):
         s = BooleanStream.true().limit(10)
         for elem in s:
@@ -13,3 +18,8 @@ def test_false(self):
         s = BooleanStream.false().limit(10)
         for elem in s:
             self.assertFalse(elem)
+
+    def test_filter(self):
+        s = BooleanStream.random().limit(20).filter(lambda x: x)
+        for elem in s:
+            self.assertTrue(elem)
