try more _spark_session

Author: bubriks

python/tests/engine/test_spark.py

@@ -1711,12 +1711,14 @@ def test_save_online_dataframe(self, mocker, backend_fixtures):
 
     def test_serialize_to_avro(self, mocker):
         # Arrange
-        SparkSession.builder \
+        spark_engine = spark.Engine()
+
+        spark_engine._spark_session.stop()
+
+        spark_engine._spark_session = SparkSession.builder \
             .config("spark.jars.packages", "org.apache.spark:spark-avro_2.12:3.1.1") \
             .getOrCreate()
 
-        spark_engine = spark.Engine()
-
         now = datetime.datetime.now()
 
         fg_data = []
@@ -1759,12 +1761,14 @@ def test_serialize_to_avro(self, mocker):
 
     def test_deserialize_from_avro(self, mocker):
         # Arrange
-        SparkSession.builder \
+        spark_engine = spark.Engine()
+
+        spark_engine._spark_session.stop()
+
+        spark_engine._spark_session = SparkSession.builder \
             .config("spark.jars.packages", "org.apache.spark:spark-avro_2.12:3.1.1") \
             .getOrCreate()
 
-        spark_engine = spark.Engine()
-
         data = []
         data.append((b"2121", b"21212121"))
         data.append((b"1212", b"12121212"))
@@ -1805,10 +1809,13 @@ def test_deserialize_from_avro(self, mocker):
 
     def test_serialize_deserialize_avro(self, mocker):
         # Arrange
-        SparkSession.builder \
+        spark_engine = spark.Engine()
+
+        spark_engine._spark_session.stop()
+
+        spark_engine._spark_session = SparkSession.builder \
             .config("spark.jars.packages", "org.apache.spark:spark-avro_2.12:3.1.1") \
             .getOrCreate()
-        spark_engine = spark.Engine()
 
         now = datetime.datetime.now()