test(core): Unify test code

j3soon · Aug 3, 2024 · f37157a · f37157a
1 parent b8aab21
commit f37157a
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 29 deletions.
diff --git a/core/tests/test_all.py b/core/tests/test_all.py
@@ -0,0 +1,15 @@
+import glob
+import os
+
+import nurse_scheduling
+
+
+def test_example_1():
+    for filepath in glob.glob("tests/testcases/*.yaml"):
+        print(f"Testing '{filepath}' ...")
+        base_filepath = os.path.splitext(filepath)[0]
+        df = nurse_scheduling.schedule(filepath, validate=False, deterministic=True)
+        print(df)
+        with open(f"{base_filepath}.csv", 'r') as f:
+            expected_csv = f.read()
+        assert df.to_csv(index=False, header=False) == expected_csv
diff --git a/core/tests/test_example_1.py b/core/tests/test_example_1.py
diff --git a/core/tests/test_or_tools_example_1.py b/core/tests/test_or_tools_example_1.py
diff --git a/core/tests/test_or_tools_example_2.py b/core/tests/test_or_tools_example_2.py