Fixed collect op when working with multiple nodes

pkg/yqlib/operator_collect_test.go

@@ -13,6 +13,13 @@ var collectOperatorScenarios = []expressionScenario{
 			"D0, P[], ()::a:\n    - 0\n",
 		},
 	},
+	{
+		skipDoc:    true,
+		expression: `[1,2,3]`,
+		expected: []string{
+			"D0, P[], (!!seq)::- 1\n- 2\n- 3\n",
+		},
+	},
 	{
 		description: "Collect empty",
 		document:    ``,
@@ -45,7 +52,8 @@ var collectOperatorScenarios = []expressionScenario{
 		expected: []string{
 			"D0, P[], (!!seq)::- cat\n",
 		},
-	}, {
+	},
+	{
 		document:   ``,
 		skipDoc:    true,
 		expression: `[true]`,
@@ -64,7 +72,7 @@ var collectOperatorScenarios = []expressionScenario{
 	{
 		document:   ``,
 		skipDoc:    true,
-		expression: `1 | collect`,
+		expression: `collect(1)`,
 		expected: []string{
 			"D0, P[], (!!seq)::- 1\n",
 		},
@@ -82,7 +90,16 @@ var collectOperatorScenarios = []expressionScenario{
 		expression: `[.a.b.[]]`,
 		skipDoc:    true,
 		expected: []string{
-			"D0, P[a b], (!!seq)::- 1\n- 2\n- 3\n",
+			"D0, P[], (!!seq)::- 1\n- 2\n- 3\n",
+		},
+	},
+	{
+		skipDoc:    true,
+		document:   `[{name: cat, thing: bor}, {name: dog}]`,
+		expression: `.[] | [.name]`,
+		expected: []string{
+			"D0, P[0], (!!seq)::- cat\n",
+			"D0, P[1], (!!seq)::- dog\n",
 		},
 	},
 }