fix coverage

kedro-org · Aug 16, 2023 · 7e06695 · 7e06695
1 parent b2062ad
commit 7e06695
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/kedro-datasets/tests/spark/test_deltatable_dataset.py b/kedro-datasets/tests/spark/test_deltatable_dataset.py
@@ -79,6 +79,9 @@ def test_exists_raises_error(self, mocker):
                 "_get_spark",
                 side_effect=AnalysisException("Other Exception", []),
             )
+        with pytest.raises(DataSetError, match="Other Exception"):
+            delta_ds.exists()
+
 
     @pytest.mark.parametrize("is_async", [False, True])
     def test_parallel_runner(self, is_async):

diff --git a/kedro-datasets/tests/spark/test_spark_dataset.py b/kedro-datasets/tests/spark/test_spark_dataset.py
@@ -419,6 +419,8 @@ def test_exists_raises_error(self, mocker):
                 "_get_spark",
                 side_effect=AnalysisException("Other Exception", []),
             )
+        with pytest.raises(DataSetError, match="Other Exception"):
+            spark_data_set.exists()
 
     @pytest.mark.parametrize("is_async", [False, True])
     def test_parallel_runner(self, is_async, spark_in):