test追加

tests/model/test_choice.py

@@ -0,0 +1,10 @@
+import pytest  # noqa
+
+from src.model.choice import Choice
+
+
+class TestChoice:
+    def test_init(self) -> None:
+        choice = Choice(name="test", odds=1.1)
+        assert choice.name == "test"
+        assert choice.odds == 1.1

tests/model/test_choice_with_bet.py

@@ -0,0 +1,16 @@
+from src.model.choice import ChoiceWithBet
+
+choice_with_bet = ChoiceWithBet(name="test", odds=1.1, bet_amount=100)
+
+
+class TestChoiceWithBet:
+    def test_init(self) -> None:
+        assert choice_with_bet.name == "test"
+        assert choice_with_bet.odds == 1.1
+        assert choice_with_bet.bet_amount == 100
+
+    def test_pay_back(self) -> None:
+        assert choice_with_bet.pay_back == 110
+
+    def test_profit(self) -> None:
+        assert choice_with_bet.profit == 10

tests/model/test_choices.py

@@ -0,0 +1,56 @@
+import pytest  # noqa
+
+from src.model.choice import Choice, Choices
+
+
+class TestChoices:
+    def test_ensure_at_least_one_choice(self) -> None:
+        with pytest.raises(ValueError):
+            Choices(root=[])
+
+    def test_from_text(self) -> None:
+        assert Choices.from_text(
+            "オッズを見て個別に選択全選択全解除\n"
+            "選択\n"
+            "1\n"
+            " \n"
+            "1\n"
+            "675.5\n"
+            "1\n"
+            "\n"
+            "2\n"
+            "93.7"
+        ).root == [
+            Choice(name="1-1", odds=675.5),
+            Choice(name="1-2", odds=93.7),
+        ]
+
+        assert Choices.from_text("\n" "1\n" "\n" "2\n" "1778.5\n").root == [
+            Choice(name="1-2", odds=1778.5),
+        ]
+
+        assert Choices.from_text(
+            "1\n"
+            "\n"
+            "2\n"
+            "\n"
+            "3\n"
+            "5688.9\n"
+            "1\n"
+            "\n"
+            "2\n"
+            "\n"
+            "4\n"
+            "9239.1\n"
+            "1\n"
+            "\n"
+            "3\n"
+            "\n"
+            "4\n"
+            "18619.1\n"
+        ).root == [
+            Choice(name="1-2-3", odds=5688.9),
+            Choice(name="1-2-4", odds=9239.1),
+            Choice(name="1-3-4", odds=18619.1),
+        ]
+        # TODO: モバイル

tests/model/test_choices_with_bet.py

@@ -0,0 +1,51 @@
+import pytest  # noqa
+
+from src.model.choice import Choice, Choices, ChoicesWithBet, ChoiceWithBet
+
+ChoiceWithBetTable = list[tuple[str, float, int, int]]
+
+choices = Choices(
+    root=[
+        Choice(name="test", odds=2.0),
+        Choice(name="test", odds=4.0),
+    ]
+)
+choices_with_bet = ChoicesWithBet(
+    root=[
+        ChoiceWithBet(name="test", odds=2.0, bet_amount=6000),
+        ChoiceWithBet(name="test", odds=4.0, bet_amount=3000),
+    ]
+)
+
+
+class TestChoicesWithBet:
+    def test_from_choice(self) -> None:
+        assert (
+            ChoicesWithBet.from_choices(choices, 9000).model_dump()
+            == choices_with_bet.model_dump()
+        )
+
+    def test_to_summary_markdwown(self) -> None:
+        # assert choices_with_bet.to_summary_markdwown() == (
+        #     "**点数**: 2 点<br>" "**購入額**: ¥ 9,000<br>" "**回収率**: 133%"
+        # )
+        assert choices_with_bet.to_summary_markdwown() == (
+            "**点数**: 2点<br>"
+            "**購入/払戻**: ¥9,000 → ¥12,000 (+3,000)<br>"
+            "**回収率**: 133%"
+        )
+
+    def test_to_table(self) -> None:
+        assert choices_with_bet.to_table() == [
+            ("test", 2.0, 6000, 12000),
+            ("test", 4.0, 3000, 12000),
+        ]
+
+    def test_total_bet_amount(self) -> None:
+        assert choices_with_bet.total_bet_amount == 9000
+
+    def test_average_pay_back(self) -> None:
+        assert choices_with_bet.average_pay_back == 12000
+
+    def test_average_profit(self) -> None:
+        assert choices_with_bet.average_profit == 3000

tests/test_func.py

@@ -1,22 +0,0 @@
-from src.func import calculate
-
-
-def test_calculate():
-    assert (
-        calculate(
-            10000,
-            (
-                "オッズを見て個別に選択全選択全解除\n"
-                "選択\n"
-                "1\n"
-                " \n"
-                "1\n"
-                "675.5\n"
-                "1\n"
-                "\n"
-                "2\n"
-                "93.7"
-            ),
-        )
-        == []
-    )