Fix array_intersect test

judahrand · judahrand · commit 4fa01adf1416 · 2024-03-04T15:57:33.000Z
diff --git a/datafusion/tests/test_functions.py b/datafusion/tests/test_functions.py
@@ -433,11 +433,11 @@ def py_flatten(arr):
             lambda: [arr[-1:2] for arr in data],
         ],
         [
-            f.array_intersect(col, [literal(3.0), literal(4.0)]),
+            f.array_intersect(col, literal([3.0, 4.0])),
             lambda: [np.intersect1d(arr, [3.0, 4.0]) for arr in data],
         ],
         [
-            f.list_intersect(col, [literal(3.0), literal(4.0)]),
+            f.list_intersect(col, literal([3.0, 4.0])),
             lambda: [np.intersect1d(arr, [3.0, 4.0]) for arr in data],
         ],
         [

Original file line number	Diff line number	Diff line change
`@@ -433,11 +433,11 @@ def py_flatten(arr):`
`433`	`433`	`lambda: [arr[-1:2] for arr in data],`
`434`	`434`	`],`
`435`	`435`	`[`
`436`		`- f.array_intersect(col, [literal(3.0), literal(4.0)]),`
	`436`	`+ f.array_intersect(col, literal([3.0, 4.0])),`
`437`	`437`	`lambda: [np.intersect1d(arr, [3.0, 4.0]) for arr in data],`
`438`	`438`	`],`
`439`	`439`	`[`
`440`		`- f.list_intersect(col, [literal(3.0), literal(4.0)]),`
	`440`	`+ f.list_intersect(col, literal([3.0, 4.0])),`
`441`	`441`	`lambda: [np.intersect1d(arr, [3.0, 4.0]) for arr in data],`
`442`	`442`	`],`
`443`	`443`	`[`