Function TestRectifyAtom

analysis/rectify_test.go:25–125

(t *testing.T)

)

func TestRectifyAtom(t *testing.T) {
	tests := []struct {
		atom      ast.Atom
		used      []ast.Variable
		wantAtom  ast.Atom
		wantFml   []ast.Term
		wantBound []ast.Variable
		fresh     []ast.Variable
	}{
		{
			atom:     atom("foo(_)"),
			wantAtom: atom("foo(_)"),
		},
		{
			atom:      atom("foo(X)"),
			wantAtom:  atom("foo(X)"),
			wantBound: []ast.Variable{ast.Variable{"X"}},
		},
		{
			atom:     atom("foo(X)"),
			used:     []ast.Variable{ast.Variable{"X"}},
			wantAtom: atom("foo(Y)"),
			wantFml:  []ast.Term{ast.Eq{ast.Variable{"Y"}, ast.Variable{"X"}}},
			fresh:    []ast.Variable{ast.Variable{"Y"}},
		},
		{
			atom:      atom("foo(X, Y)"),
			used:      []ast.Variable{ast.Variable{"X"}},
			wantAtom:  atom("foo(Z, Y)"),
			wantFml:   []ast.Term{ast.Eq{ast.Variable{"Z"}, ast.Variable{"X"}}},
			wantBound: []ast.Variable{ast.Variable{"Y"}},
			fresh:     []ast.Variable{ast.Variable{"Z"}},
		},
		{
			atom:     atom("foo(X, 23001)"),
			used:     []ast.Variable{ast.Variable{"X"}},
			wantAtom: atom("foo(Z, Y)"),
			wantFml: []ast.Term{
				ast.Eq{ast.Variable{"Z"}, ast.Variable{"X"}},
				ast.Eq{ast.Variable{"Y"}, ast.Number(23001)}},
			fresh: []ast.Variable{ast.Variable{"Z"}, ast.Variable{"Y"}},
		},
		{
			atom:      atom("foo(X, X)"),
			wantAtom:  atom("foo(X, Y)"),
			wantFml:   []ast.Term{ast.Eq{ast.Variable{"Y"}, ast.Variable{"X"}}},
			wantBound: []ast.Variable{ast.Variable{"X"}},
			fresh:     []ast.Variable{ast.Variable{"Y"}},
		},
		{
			atom:     atom("foo(fn:plus(X, 1))"),
			wantAtom: atom("foo(Y)"),
			wantFml:  []ast.Term{fml("Y = fn:plus(X, 1)")},
			fresh:    []ast.Variable{ast.Variable{"Y"}},
		},
		{
			atom:     atom("foo(fn:plus(A, fn:plus(B, 1)))"),
			wantAtom: atom("foo(Y)"),