39_golf: Python implementation

39_Golf/python/test_golf.py

@@ -0,0 +1,88 @@
+import io
+import math
+
+import pytest
+from golf import (
+    CircleGameObj,
+    GameObjType,
+    Golf,
+    Point,
+    RectGameObj,
+    get_distance,
+    is_in_rectangle,
+    odds,
+    to_degrees_360,
+    to_radians,
+)
+
+
+def test_odds():
+    n = 1000
+    p = sum(odds(50) for i in range(n)) / n
+    assert abs(p - 0.5) < 0.1
+
+
+@pytest.mark.parametrize(
+    ("p1", "p2", "expected"),
+    [
+        ((0, 0), (0, 0), 0),
+        ((0, 0), (1, 0), 1),
+        ((0, 0), (0, 1), 1),
+        ((0, 1), (0, 0), 1),
+        ((1, 0), (0, 0), 1),
+        ((0, 0), (2, 0), 2),
+        ((0, 0), (0, 2), 2),
+        ((0, 2), (0, 0), 2),
+        ((2, 0), (0, 0), 2),
+        ((0, 0), (1, 1), 2**0.5),
+        ((2, 3), (4, 5), (2**2 + 2**2) ** 0.5),
+    ],
+)
+def test_get_distance(p1, p2, expected):
+    assert get_distance(Point(*p1), Point(*p2)) == expected
+
+
+@pytest.mark.parametrize(
+    ("pt", "rect", "expected"),
+    [
+        (
+            CircleGameObj(1, 1, 1, GameObjType.BALL),
+            RectGameObj(0, 0, 2, 2, GameObjType.GREEN),
+            True,
+        ),
+        (
+            CircleGameObj(1, 1, 1, GameObjType.BALL),
+            RectGameObj(0, 0, 1, 1, GameObjType.GREEN),
+            False,
+        ),
+    ],
+)
+def test_is_in_rectangle(pt, rect, expected):
+    assert is_in_rectangle(pt, rect) == expected
+
+
+@pytest.mark.parametrize(
+    ("angle", "radians"),
+    [
+        (0, 0),
+        (180, math.pi),
+        (360, 2 * math.pi),
+    ],
+)
+def test_to_radians(angle, radians):
+    assert to_radians(angle) == radians
+    assert to_degrees_360(radians) == angle
+
+
+def test_golf(monkeypatch, capsys):
+    handycap = 10
+    difficulty = 4
+    club = 10
+    swing = 10
+    monkeypatch.setattr(
+        "sys.stdin",
+        io.StringIO(f"10\na\n{handycap}\n{difficulty}\n{club}\n{swing}\nQUIT"),
+    )
+    Golf()
+    out, err = capsys.readouterr()
+    assert err == ""