1 files changed, 46 insertions, 0 deletions
diff --git a/tests/models/thl/test_payout_format.py b/tests/models/thl/test_payout_format.py
new file mode 100644
index 0000000..dc91f39
--- /dev/null
+++ b/tests/models/thl/test_payout_format.py
@@ -0,0 +1,46 @@
+import pytest
+from pydantic import BaseModel
+
+from generalresearch.models.thl.payout_format import (
+    PayoutFormatType,
+    PayoutFormatField,
+    format_payout_format,
+)
+
+
+class PayoutFormatTestClass(BaseModel):
+    payout_format: PayoutFormatType = PayoutFormatField
+
+
+class TestPayoutFormat:
+    def test_payout_format_cls(self):
+        # valid
+        PayoutFormatTestClass(payout_format="{payout*10:,.0f} Points")
+        PayoutFormatTestClass(payout_format="{payout:.0f}")
+        PayoutFormatTestClass(payout_format="${payout/100:.2f}")
+
+        # invalid
+        with pytest.raises(expected_exception=Exception) as e:
+            PayoutFormatTestClass(payout_format="{payout10:,.0f} Points")
+
+        with pytest.raises(expected_exception=Exception) as e:
+            PayoutFormatTestClass(payout_format="payout:,.0f} Points")
+
+        with pytest.raises(expected_exception=Exception):
+            PayoutFormatTestClass(payout_format="payout")
+
+        with pytest.raises(expected_exception=Exception):
+            PayoutFormatTestClass(payout_format="{payout;import sys:.0f}")
+
+    def test_payout_format(self):
+        assert "1,230 Points" == format_payout_format(
+            payout_format="{payout*10:,.0f} Points", payout_int=123
+        )
+
+        assert "123" == format_payout_format(
+            payout_format="{payout:.0f}", payout_int=123
+        )
+
+        assert "$1.23" == format_payout_format(
+            payout_format="${payout/100:.2f}", payout_int=123
+        )