aboutsummaryrefslogtreecommitdiff
path: root/tests/test_models.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_models.py')
-rw-r--r--tests/test_models.py28
1 files changed, 28 insertions, 0 deletions
diff --git a/tests/test_models.py b/tests/test_models.py
new file mode 100644
index 0000000..37e9c03
--- /dev/null
+++ b/tests/test_models.py
@@ -0,0 +1,28 @@
+import pytest
+from pydantic import ValidationError
+
+from poker.constants import Suit, Value
+from poker.models import Card, Hand
+
+
+def test_hand_should_contain_unique_cards() -> None:
+ with pytest.raises(ValidationError):
+ _ = Hand(
+ cards=(
+ Card(suit=Suit.CLUBS, value=Value.ACE),
+ Card(suit=Suit.CLUBS, value=Value.TWO),
+ Card(suit=Suit.CLUBS, value=Value.THREE),
+ Card(suit=Suit.CLUBS, value=Value.FOUR),
+ Card(suit=Suit.CLUBS, value=Value.FOUR),
+ )
+ )
+
+ _ = Hand(
+ cards=(
+ Card(suit=Suit.CLUBS, value=Value.ACE),
+ Card(suit=Suit.CLUBS, value=Value.TWO),
+ Card(suit=Suit.CLUBS, value=Value.THREE),
+ Card(suit=Suit.CLUBS, value=Value.FOUR),
+ Card(suit=Suit.CLUBS, value=Value.FIVE),
+ )
+ )