add EvalPlayer

Author: lars20070

pydantic_evals/pydantic_evals/tournament.py

@@ -0,0 +1,10 @@
+from __future__ import annotations as _annotations
+
+from pydantic import BaseModel, Field
+
+
+class EvalPlayer(BaseModel):
+    """Player in a Bradley-Terry tournament."""
+    idx: int = Field(..., description='unique identifier for the player')
+    item: str = Field(..., description='item to be scored')
+    score: float | None = Field(default=None, description='Bradley-Terry strength score for the item')

tests/evals/test_tournament.py

@@ -0,0 +1,19 @@
+from __future__ import annotations as _annotations
+
+from ..conftest import try_import
+
+with try_import() as imports_successful:
+    from pydantic_evals.tournament import EvalPlayer
+
+
+    def test_evalplayer() -> None:
+        """
+        Test the EvalPlayer class.
+        """
+        player = EvalPlayer(
+            idx=42,
+            item='toasted rice & miso caramel ice cream',
+        )
+        assert player.idx == 42
+        assert player.item == 'toasted rice & miso caramel ice cream'
+