mymax: step 03

ramalho · ramalho · commit 08c1a9a6690b · 2021-05-09T22:46:47.000-03:00
diff --git a/less_than/max_steps/03-two_numbers-TypeVar/errors.py b/less_than/max_steps/03-two_numbers-TypeVar/errors.py
@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+
+from typing import TYPE_CHECKING, List, Optional
+
+import mymax as my
+
+def demo_args_float() -> None:
+    a, b = 2.5, 3.5
+    expected = b
+    result = my.max(a, b)
+    assert result == expected
+    print((a, b), '->', result)
+
+
+def demo_args_fraction() -> None:
+    a, b = Fraction(5, 8), Fraction(7, 12)
+    expected = a
+    result = my.max(a, b)
+    assert result == expected
+    print((a, b), '->', result)
+    print(result.numerator, result.denominator, sep='/')
+    if TYPE_CHECKING:
+        reveal_type(a)
+        reveal_type(b)
+        reveal_type(result)
+
+
+def demo_args_fraction_float() -> None:
+    a, b = Fraction(5, 8), .5
+    expected = a
+    result = my.max(a, b)
+    assert result == expected
+    print((a, b), '->', result)
+    print(result.numerator, result.denominator, sep='/')
+    if TYPE_CHECKING:
+        reveal_type(a)
+        reveal_type(b)
+        reveal_type(result)
+
+
+def demo_args_float_fraction() -> None:
+    a, b = .9, Fraction(5, 8)
+    expected = a
+    result = my.max(a, b)
+    assert result == expected
+    print((a, b), '->', result)
+    print(result.numerator, result.denominator, sep='/')
+    if TYPE_CHECKING:
+        reveal_type(a)
+        reveal_type(b)
+        reveal_type(result)
+
+
+###################################### intentional type errors
+
+from fractions import Fraction
+
+
+
+def main():
+    import sys
+    for name, val in globals().items():
+        if name.startswith('demo') or name.startswith('error'):
+            print('_' * 20, name)
+            val()
+
+if __name__ == '__main__':
+    main()
diff --git a/less_than/max_steps/03-two_numbers-TypeVar/fixed.py b/less_than/max_steps/03-two_numbers-TypeVar/fixed.py
@@ -12,11 +12,7 @@ def demo_args_float() -> None:
     print((a, b), '->', result)
 
 
-###################################### intentional type errors
-
-from fractions import Fraction
-
-def error_args_fraction() -> None:
+def demo_args_fraction() -> None:
     a, b = Fraction(5, 8), Fraction(7, 12)
     expected = a
     result = my.max(a, b)
@@ -29,6 +25,11 @@ def error_args_fraction() -> None:
         reveal_type(result)
 
 
+###################################### intentional type errors
+
+from fractions import Fraction
+
+
 
 def main():
     import sys
