Add pytest

pyproject.toml

@@ -13,6 +13,9 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.12"
 
+[tool.poetry.group.test.dependencies]
+pytest = "^8.2.2"
+
 [tool.poetry.group.lint.dependencies]
 pre-commit = "^3.7.1"
 ruff = "^0.4.6"

tests/test_model_creation.py

@@ -0,0 +1,23 @@
+# content of test_sysexit.py
+import stormvogel.model
+
+
+def test_mdp_creation():
+    dtmc = stormvogel.model.new_dtmc("Die")
+
+    init = dtmc.get_initial_state()
+
+    # roll die
+    init.set_transitions(
+        [(1 / 6, dtmc.new_state(f"rolled{i}", {"rolled": i})) for i in range(6)]
+    )
+
+    assert len(dtmc.states) == 7
+    assert len(dtmc.transitions) == 7
+    # Check that all states 1..6 have self loops
+    for i in range(1, 7):
+        # yeah we need transition getting syntax
+        assert (
+            dtmc.transitions[i].transition[stormvogel.model.EmptyAction].branch[0][1].id
+            == i
+        )