Updating a unit test in test_utils.py

HopkinsIDD · Nov 20, 2024 · 918ebc7 · 918ebc7
1 parent 97d3996
commit 918ebc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flepimop/gempyor_pkg/tests/utils/test_utils.py b/flepimop/gempyor_pkg/tests/utils/test_utils.py
@@ -37,7 +37,7 @@ def test_read_df_and_write_success(fname, extension):
     ("fname", "extension"), [("mobility", "csv"), ("usa-geoid-params-output", "parquet")]
 )
 def test_read_df_and_write_fail(fname, extension):
-    with pytest.raises(NotImplementedError, match=r".*Invalid.*extension.*Must.*"):
+    with pytest.raises(ValueError, match=r".*Invalid.*extension.*Must.*"):
         os.chdir(tmp_path)
         os.makedirs("data", exist_ok=True)
         os.chdir("data")