test

bubriks · bubriks · commit 3e0dc475ea39 · 2024-09-23T20:09:13.000+03:00
diff --git a/python/tests/engine/test_spark.py b/python/tests/engine/test_spark.py
@@ -4559,7 +4559,7 @@ def test_save_empty_dataframe(self, mocker):
         mock_spark_engine_save_dataframe = mocker.patch(
             "hsfs.engine.spark.Engine.save_dataframe"
         )
-        mock_spark_table = mocker.patch("pyspark.sql.session.SparkSession.read")
+        mock_spark_table = mocker.patch("pyspark.sql.session.SparkSession.read.format")
 
         # Arrange
         spark_engine = spark.Engine()

Original file line number	Diff line number	Diff line change
`@@ -4559,7 +4559,7 @@ def test_save_empty_dataframe(self, mocker):`
`4559`	`4559`	`mock_spark_engine_save_dataframe = mocker.patch(`
`4560`	`4560`	`"hsfs.engine.spark.Engine.save_dataframe"`
`4561`	`4561`	`)`
`4562`		`- mock_spark_table = mocker.patch("pyspark.sql.session.SparkSession.read")`
	`4562`	`+ mock_spark_table = mocker.patch("pyspark.sql.session.SparkSession.read.format")`
`4563`	`4563`
`4564`	`4564`	`# Arrange`
`4565`	`4565`	`spark_engine = spark.Engine()`