remove fixture

databricks · Mar 5, 2024 · 289839b · 289839b
1 parent 7e2f571
commit 289839b
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/libs/template/templates/default-python/template/{{.project_name}}/tests/main_test.py.tmpl b/libs/template/templates/default-python/template/{{.project_name}}/tests/main_test.py.tmpl
@@ -1,15 +1,7 @@
 from pyspark.sql import SparkSession
-from pytest import fixture
 from {{.project_name}}.main import get_taxis, get_spark
 
 
-@fixture(scope="session")
-def spark() -> SparkSession:
-    spark = get_spark()
-    yield spark
-    spark.stop()
-
-
-def test_main(spark: SparkSession):
-    taxis = get_taxis(spark)
+def test_main():
+    taxis = get_taxis(get_spark())
     assert taxis.count() > 5