Regex update in test_write_df.py

HopkinsIDD · Dec 3, 2024 · 8a5e772 · 8a5e772
1 parent 8255da2
commit 8a5e772
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flepimop/gempyor_pkg/tests/utils/test_write_df.py b/flepimop/gempyor_pkg/tests/utils/test_write_df.py
@@ -30,13 +30,13 @@ def test_raises_not_implemented_error(self) -> None:
         """
         with pytest.raises(
             expected_exception=NotImplementedError,
-            match=r".*Supported extensions are `.csv` or `.parquet`.",
+            match=r"Invalid extension provided: '.txt'. Supported extensions are `.csv` or `.parquet`.",
         ) as _:
             with NamedTemporaryFile(suffix=".txt") as temp_file:
                 write_df(fname=temp_file.name, df=self.sample_df)
         with pytest.raises(
             expected_exception=NotImplementedError,
-            match=r".*Supported extensions are `.csv` or `.parquet`.",
+            match=r"Invalid extension provided: '.txt'. Supported extensions are `.csv` or `.parquet`.",
         ) as _:
             with NamedTemporaryFile(suffix=".txt") as temp_file:
                 fname = temp_file.name[:-4]