add tests for check

mishankov · Feb 8, 2024 · 778a79c · 778a79c
1 parent ef2decb
commit 778a79c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_tipe.py b/tests/test_tipe.py
@@ -22,3 +22,14 @@ def add(a: int, b: int) -> int:
 
     assert Pipe(1).pipe(add, 2).unwrap() == 3
     assert Pipe(2).pipe(range, 4).pipe(len).unwrap() == 2
+
+
+def test_check_run(capsys):
+    assert Pipe(1).check(print).unwrap() == 1
+
+    captured = capsys.readouterr()
+    assert captured.out == "1\n"
+
+
+def test_check_immutability():
+    assert Pipe(1).check(lambda x: x + 1).unwrap() == 1