moved fixture to respective module

Author: mwojtyczka

tests/integration/conftest.py

@@ -1,4 +1,3 @@
-import os
 import logging
 
 from pytest import fixture
@@ -21,8 +20,4 @@ def product_info():
     return 'pytester', __version__
 
 
-@fixture
-def serverless_env():
-    os.environ['DATABRICKS_SERVERLESS_COMPUTE_ID'] = 'auto'
-    yield
-    os.environ.pop('DATABRICKS_SERVERLESS_COMPUTE_ID')
+

tests/integration/fixtures/test_connect.py

@@ -1,8 +1,19 @@
+import os
+from pytest import fixture
+
+
 def test_databricks_connect(spark):
     rows = spark.sql("SELECT 1").collect()
     assert rows[0][0] == 1
 
 
-def test_databricks_connect_serverless(serverless_env, spark) -> None:
+@fixture
+def serverless_env():
+    os.environ['DATABRICKS_SERVERLESS_COMPUTE_ID'] = 'auto'
+    yield
+    os.environ.pop('DATABRICKS_SERVERLESS_COMPUTE_ID')
+
+
+def test_databricks_connect_serverless(serverless_env, spark):
     rows = spark.sql("SELECT 1").collect()
     assert rows[0][0] == 1