Add tests for Try

Author: dbrattli

expression/core/try_.py

@@ -21,6 +21,14 @@ class Try(Result[_TSource, Exception]):
     to use when the error type is an exception.
     """
 
+    def __str__(self) -> str:
+        """Return a string representation of the Try."""
+        match self:
+            case Try(tag="ok", ok=ok):
+                return f"Success {ok}"
+            case Try(error=error):
+                return f"Failure {error}"
+
 
 def Success(value: _TSource) -> Try[_TSource]:
     """The successful Try case.

tests/test_gen.py

@@ -114,7 +114,7 @@ def fn() -> Generator[int, None, None]:
     assert value == 42
 
 
-def test_generator_with_yield_from_gen():
+def test_generator_with_yield_from_subgen():
     def gn():
         yield 42
 
@@ -137,3 +137,24 @@ def fn() -> Generator[int, None, None]:
     gen = fn()
     value = next(gen)
     assert value == 42
+
+def test_generator_with_yield_from_send_to_subgen():
+    def gn() -> Generator[int, int|None, int]:
+        ret = yield 42
+        print("subgenerator: ret", ret)
+        assert ret is not None
+        return ret + 1
+
+    def fn() -> Generator[int, int|None, int]:
+        ret = yield from gn()
+        print("generator: ret", ret)
+        return ret +1
+
+    gen = fn()
+    value = gen.send(None)
+    assert value == 42
+    try:
+        value = gen.send(10)
+    except StopIteration as ex:
+        print("StopIteration", ex.value)
+        assert ex.value == 12

tests/test_try.py

@@ -28,3 +28,12 @@ def test_try_match_failure():
             assert err == error
         case _:
             assert False
+
+def test_try_to_string_success():
+    xs: Try[int] = Success(10)
+    assert str(xs) == "Success 10"
+
+def test_try_to_string_failure():
+    error = Exception("err")
+    xs: Try[int] = Failure(error)
+    assert str(xs) == "Failure err"