Function TestComprehension

common/ast/expr_test.go:186–239

(t *testing.T)

}

func TestComprehension(t *testing.T) {
	fac := ast.NewExprFactory()
	expr := fac.NewComprehension(1,
		fac.NewList(2, []ast.Expr{
			fac.NewLiteral(3, types.Int(1)),
			fac.NewLiteral(4, types.Int(2)),
		}, []int32{}),
		"i",
		"__result__",
		fac.NewLiteral(5, types.False),
		fac.NewLiteral(6, types.True),
		fac.NewCall(7, "_||_",
			fac.NewAccuIdent(8),
			fac.NewCall(9, "<", fac.NewIdent(10, "i"), fac.NewLiteral(11, types.Int(3))),
		),
		fac.NewAccuIdent(12),
	)
	comp := expr.AsComprehension()
	if comp.IterRange().Kind() != ast.ListKind {
		t.Errorf("IterRange() got %v, wanted list", comp.IterRange().Kind())
	}
	if comp.AccuInit().Kind() != ast.LiteralKind {
		t.Errorf("AccuInit() got %v, wanted literal", comp.AccuInit().Kind())
	}
	if comp.LoopCondition().Kind() != ast.LiteralKind {
		t.Errorf("LoopCondition() got %v, wanted literal", comp.LoopCondition().Kind())
	}
	if comp.LoopStep().Kind() != ast.CallKind {
		t.Errorf("LoopStep() got %v, wanted call", comp.LoopStep().Kind())
	}
	if comp.Result().Kind() != ast.IdentKind {
		t.Errorf("Result() got %v, wanted identifier", comp.Result().Kind())
	}
	expr.RenumberIDs(testIDGen(100))
	if expr.ID() != 101 {
		t.Errorf("Renumbering an unspecified expression mutated the value: %v", expr)
	}
	comp = expr.AsComprehension()
	if comp.IterRange().ID() != 102 {
		t.Errorf("IterRange() got %v, wanted list", comp.IterRange().ID())
	}
	if comp.AccuInit().ID() != 105 {
		t.Errorf("AccuInit() got %v, wanted literal", comp.AccuInit().ID())
	}
	if comp.LoopCondition().ID() != 106 {
		t.Errorf("LoopCondition() got %v, wanted literal", comp.LoopCondition().ID())
	}
	if comp.LoopStep().ID() != 107 {
		t.Errorf("LoopStep() got %v, wanted call", comp.LoopStep().ID())
	}
	if comp.Result().ID() != 112 {
		t.Errorf("Result() got %v, wanted identifier", comp.Result().ID())
	}
}

func TestComprehensionNil(t *testing.T) {
	expr := nilTestExpr(t)