refactor: Simplify df fixture by passing ctx as an argument

kosiew · kosiew · commit 3eade9586a9f · 2025-03-13T17:37:47.000+08:00
diff --git a/python/tests/test_udwf.py b/python/tests/test_udwf.py
@@ -182,9 +182,7 @@ def ctx():
 
 
 @pytest.fixture
-def df():
-    ctx = SessionContext()
-
+def df(ctx):
     # create a RecordBatch and a new DataFrame from it
     batch = pa.RecordBatch.from_arrays(
         [

Original file line number	Diff line number	Diff line change
`@@ -182,9 +182,7 @@ def ctx():`
`182`	`182`
`183`	`183`
`184`	`184`	`@pytest.fixture`
`185`		`-def df():`
`186`		`- ctx = SessionContext()`
`187`		`-`
	`185`	`+def df(ctx):`
`188`	`186`	`# create a RecordBatch and a new DataFrame from it`
`189`	`187`	`batch = pa.RecordBatch.from_arrays(`
`190`	`188`	`[`