Fix array_union test

judahrand · judahrand · commit 76ad39d432aa · 2024-03-04T15:58:45.000Z
diff --git a/datafusion/tests/test_functions.py b/datafusion/tests/test_functions.py
@@ -441,11 +441,11 @@ def py_flatten(arr):
             lambda: [np.intersect1d(arr, [3.0, 4.0]) for arr in data],
         ],
         [
-            f.array_union(col, [literal(12.0), literal(999.0)]),
+            f.array_union(col, literal([12.0, 999.0])),
             lambda: [np.union1d(arr, [12.0, 999.0]) for arr in data],
         ],
         [
-            f.list_union(col, [literal(12.0), literal(999.0)]),
+            f.list_union(col, literal([12.0, 999.0])),
             lambda: [np.union1d(arr, [12.0, 999.0]) for arr in data],
         ],
         [

Original file line number	Diff line number	Diff line change
`@@ -441,11 +441,11 @@ def py_flatten(arr):`
`441`	`441`	`lambda: [np.intersect1d(arr, [3.0, 4.0]) for arr in data],`
`442`	`442`	`],`
`443`	`443`	`[`
`444`		`- f.array_union(col, [literal(12.0), literal(999.0)]),`
	`444`	`+ f.array_union(col, literal([12.0, 999.0])),`
`445`	`445`	`lambda: [np.union1d(arr, [12.0, 999.0]) for arr in data],`
`446`	`446`	`],`
`447`	`447`	`[`
`448`		`- f.list_union(col, [literal(12.0), literal(999.0)]),`
	`448`	`+ f.list_union(col, literal([12.0, 999.0])),`
`449`	`449`	`lambda: [np.union1d(arr, [12.0, 999.0]) for arr in data],`
`450`	`450`	`],`
`451`	`451`	`[`