adding method definition

william-conti · JCZuurmond · commit 3e10dc146c63 · 2025-02-26T10:54:17.000+01:00
diff --git a/tests/unit/source_code/linters/test_pyspark.py b/tests/unit/source_code/linters/test_pyspark.py
@@ -527,7 +527,10 @@ def test_from_table_sql_py_linter_fixes_migrated_table(migration_index) -> None:
         ),
         # Test for mount calls inside a function
         (
-            """variable = searchPath('/mnt/something/', some_variable)""",
+            """
+            def searchPath(rep, dateAlternative = None):
+                return dbutils.fs.ls(rep)
+            variable = searchPath('/mnt/something/', some_variable)""",
             [
                 Deprecation(
                     code='direct-filesystem-access',

Original file line number	Diff line number	Diff line change
`@@ -527,7 +527,10 @@ def test_from_table_sql_py_linter_fixes_migrated_table(migration_index) -> None:`
`527`	`527`	`),`
`528`	`528`	`# Test for mount calls inside a function`
`529`	`529`	`(`
`530`		`- """variable = searchPath('/mnt/something/', some_variable)""",`
	`530`	`+ """`
	`531`	`+ def searchPath(rep, dateAlternative = None):`
	`532`	`+ return dbutils.fs.ls(rep)`
	`533`	`+ variable = searchPath('/mnt/something/', some_variable)""",`
`531`	`534`	`[`
`532`	`535`	`Deprecation(`
`533`	`536`	`code='direct-filesystem-access',`