fix: test case

GoogleCloudPlatform · Nov 13, 2024 · 8f4ec6a · 8f4ec6a
1 parent e66d927
commit 8f4ec6a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/test/gcs/test_gcs_to_bigtable.py b/python/test/gcs/test_gcs_to_bigtable.py
@@ -61,7 +61,7 @@ def test_run(self, mock_spark_session):
         mock_spark_session.dataframe.DataFrame.write.format. \
             assert_called_once_with(constants.FORMAT_BIGTABLE)
         mock_spark_session.dataframe.DataFrame.write.format().options. \
-            assert_called_with(catalog='gs://catalog/catalog.json')
+            assert_called_with(catalog='{key:value}')
 
     @mock.patch.object(pyspark.sql, 'SparkSession')
     def test_run_csv1(self, mock_spark_session):
@@ -88,7 +88,7 @@ def test_run_csv1(self, mock_spark_session):
         mock_spark_session.dataframe.DataFrame.write.format. \
             assert_called_once_with(constants.FORMAT_BIGTABLE)
         mock_spark_session.dataframe.DataFrame.write.format().options. \
-            assert_called_with(catalog='gs://catalog/catalog.json')
+            assert_called_with(catalog='{key:value}')
 
     @mock.patch.object(pyspark.sql, 'SparkSession')
     def test_run_csv2(self, mock_spark_session):
@@ -121,4 +121,4 @@ def test_run_csv2(self, mock_spark_session):
         mock_spark_session.dataframe.DataFrame.write.format. \
             assert_called_once_with(constants.FORMAT_BIGTABLE)
         mock_spark_session.dataframe.DataFrame.write.format().options. \
-            assert_called_with(catalog='gs://catalog/catalog.json')
+            assert_called_with(catalog='{key:value}')