more python tests

flexatone · Apr 8, 2024 · ceb06e0 · ceb06e0
1 parent a52c32c
commit ceb06e0
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_sieve.py b/tests/test_sieve.py
@@ -11,12 +11,30 @@ def test_sieve_contains_a() -> None:
     assert 10 in s1
     assert 15 in s1
 
+def test_sieve_invert_a() -> None:
+    s1 = Sieve("5@0")
+    s2 = ~s1
+    assert str(s3) == "Sieve{!5@0}"
+    assert 4 in s1
+
 def test_sieve_xor_a() -> None:
     s1 = Sieve("3@2")
     s2 = Sieve("5@1")
     s3 = s1 ^ s2
     assert str(s3) == "Sieve{3@2^5@1}"
 
+def test_sieve_or_a() -> None:
+    s1 = Sieve("3@2")
+    s2 = Sieve("5@1")
+    s3 = s1 | s2
+    assert str(s3) == "Sieve{3@2|5@1}"
+
+def test_sieve_and_a() -> None:
+    s1 = Sieve("3@2")
+    s2 = Sieve("5@1")
+    s3 = s1 & s2
+    assert str(s3) == "Sieve{3@2&5@1}"
+