python version of exercise

python/tests/test_statement.py

@@ -0,0 +1,25 @@
+import json
+
+import pytest
+from approvaltests import verify
+from approvaltests.utils import get_adjacent_file
+
+from statement import statement
+
+
+def test_example_statement():
+    with open(get_adjacent_file("invoice.json")) as f:
+        invoice = json.loads(f.read())
+    with open(get_adjacent_file("plays.json")) as f:
+        plays = json.loads(f.read())
+    verify(statement(invoice, plays))
+
+
+def test_statement_with_new_play_types():
+    with open(get_adjacent_file("invoice_new_plays.json")) as f:
+        invoice = json.loads(f.read())
+    with open(get_adjacent_file("new_plays.json")) as f:
+        plays = json.loads(f.read())
+    with pytest.raises(ValueError) as exception_info:
+        statement(invoice, plays)
+    assert "unknown type" in str(exception_info.value)