def test_constraint_forAll2(self):
# !x.(P=>Q)
# Build AST:
string_to_file("#PREDICATE f={(1,7),(2,8),(3,9)} & S={1,2,3} & !(x,y).(y:INTEGER &(x:S & f(x)<y) & y<42 =>y:T)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Test
env = Environment()
lst = [("S", PowerSetType(IntegerType())),("f", PowerSetType(CartType(PowerSetType(IntegerType()), PowerSetType(IntegerType()))))]
type_with_known_types(root, env, lst, ["T"])
assert isinstance(get_type_by_name(env, "x"), IntegerType)
assert isinstance(get_type_by_name(env, "y"), IntegerType)
assert isinstance(get_type_by_name(env, "T"), PowerSetType)
assert isinstance(get_type_by_name(env, "T").data, IntegerType)
env.add_ids_to_frame(["f","S","T"])
env.set_value("f", frozenset([(1,7),(2,8),(3,9)]))
env.set_value("S", frozenset([1,2,3]))
env.set_value("T", frozenset(range(10,42)))
env._min_int = -2**8
env._max_int = 2**8
unqantPred = root.children[0].children[1]
assert isinstance(unqantPred, AForallPredicate)
varList = unqantPred.children[0:-1]
P = unqantPred.children[-1].children[0]
Q = unqantPred.children[-1].children[1]
domain = compute_using_external_solver(P, env, varList)
assert frozenset([x["x"] for x in domain])==frozenset([1,2,3])
domain = compute_using_external_solver(P, env, varList)
assert frozenset([x["y"] for x in domain])==frozenset(range(8,42))
def test_library_length(self):
string = '''
MACHINE LibraryStrings
CONSTANTS length
PROPERTIES
/* compute the length of a string */
length: STRING --> INTEGER &
length = %x.(x:STRING|STRING_LENGTH(x))
DEFINITIONS
STRING_LENGTH(x) == length(x);
EXTERNAL_FUNCTION_STRING_LENGTH == STRING --> INTEGER;
ASSERTIONS
length("abc") = 3;
length("") = 0;
length("hello") = 5
END
'''
# Build AST
string_to_file(string, file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Test
env = Environment()
dh = DefinitionHandler(env, str_ast_to_python_ast)
dh.repl_defs(root)
mch = parse_ast(root, env)
type_check_bmch(root, env, mch)
assert isinstance(get_type_by_name(env, "length"), PowerSetType)
assert isinstance(get_type_by_name(env, "length").data, CartType)
assert isinstance(get_type_by_name(env, "length").data.left.data, StringType)
assert isinstance(get_type_by_name(env, "length").data.right.data, IntegerType)
arbitrary_init_machine(root, env, mch) # init VARIABLES and eval INVARIANT
assert isinstance(root.children[4], AAssertionsMachineClause)
interpret(root.children[4], env)
def test_types_expr_leq(self):
# Build AST
string_to_file("#PREDICATE x<y & y=1", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["x","y"])
assert isinstance(get_type_by_name(env, "x"), IntegerType)
assert isinstance(get_type_by_name(env, "y"), IntegerType)
def test_types_simple_mul_unify2(self):
# Build AST
string_to_file("#PREDICATE 42=a*b", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["b","a"])
assert isinstance(get_type_by_name(env, "a"), IntegerType)
assert isinstance(get_type_by_name(env, "b"), IntegerType)
def test_types_interval2(self):
# Build AST:
string_to_file("#PREDICATE x=1..5", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["x"])
assert isinstance(get_type_by_name(env, "x"), PowerSetType)
assert isinstance(get_type_by_name(env, "x").data, IntegerType)
def test_types_function_app2(self):
# Build AST
string_to_file('#PREDICATE f= {1 |-> "aa", 2 |-> "bb"}(xx) & xx=1 ', file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["xx", "f"])
assert isinstance(get_type_by_name(env, "f"), StringType)
assert isinstance(get_type_by_name(env, "xx"), IntegerType)
def test_types_simple_set_com(self):
# Build AST
string_to_file("#PREDICATE ID={x|x<10}", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["ID"])
assert isinstance(get_type_by_name(env, "ID"), PowerSetType)
assert isinstance(get_type_by_name(env, "x"), IntegerType)
def test_types_set_power_unify2(self):
# Build AST
string_to_file("#PREDICATE 1:S ", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["S"])
assert isinstance(get_type_by_name(env, "S"), PowerSetType)
assert isinstance(get_type_by_name(env, "S").data, IntegerType)
def type_check_sequence(self, ast_string):
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X")))]
type_with_known_types(root, env, lst, ["s"])
assert isinstance(get_type_by_name(env, "s"), PowerSetType)
assert isinstance(get_type_by_name(env, "s").data, CartType)
assert isinstance(get_type_by_name(env, "s").data.left.data, IntegerType)
assert isinstance(get_type_by_name(env, "s").data.right.data, SetType)
def test_types_simple_pow(self):
# Build AST
string_to_file("#PREDICATE S=POW(B)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("B", PowerSetType(SetType("X")))]
type_with_known_types(root, env, lst, ["S"])
assert isinstance(get_type_by_name(env, "S"), PowerSetType)
assert isinstance(get_type_by_name(env, "S").data, PowerSetType)
def test_types_simple_rev(self):
# Build AST
string_to_file("#PREDICATE r:A<->B & f = r~ & x:dom(f)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("A", PowerSetType(SetType("X"))),("B", PowerSetType(SetType("Y")))]
type_with_known_types(root, env, lst, ["r","x","f"])
assert isinstance(get_type_by_name(env, "x"), SetType)
assert get_type_by_name(env, "x").name =="Y"
def test_types_seq_last(self):
# Build AST
string_to_file("#PREDICATE s:perm(S) & n=last(s)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X")))]
type_with_known_types(root, env, lst, ["s", "n"])
assert isinstance(get_type_by_name(env, "n"), SetType)
assert get_type_by_name(env, "n").name == "X"
def test_types_simple_sub_res(self):
# Build AST
string_to_file("#PREDICATE r:A<->B & v=S <| r", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X"))),("A", PowerSetType(SetType("X"))),("B", PowerSetType(SetType("Y")))]
type_with_known_types(root, env, lst, ["r","v"])
assert isinstance(get_type_by_name(env, "v"), PowerSetType)
assert isinstance(get_type_by_name(env, "v").data, CartType)
def test_types_bool(self):
# Build AST
string_to_file("#PREDICATE A:BOOL & B:BOOL & C:BOOL & (A=TRUE <=> (B=FALSE or C=FALSE)) & (B=TRUE <=> A=TRUE)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["A","B","C"])
assert isinstance(get_type_by_name(env, "A"), BoolType)
assert isinstance(get_type_by_name(env, "B"), BoolType)
assert isinstance(get_type_by_name(env, "C"), BoolType)
def test_types_function_app(self):
# Build AST
string_to_file("#PREDICATE f:S+->T & y=f(x)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("x", SetType("X")), ("S", PowerSetType(SetType("X"))), ("T", PowerSetType(SetType("Y")))]
type_with_known_types(root, env, lst, ["y", "f"])
assert isinstance(get_type_by_name(env, "y"), SetType)
assert get_type_by_name(env, "y").name == "Y"
def test_types_range(self):
# Build AST
string_to_file("#PREDICATE r:S<->T & x:ran(r)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X"))),("T", PowerSetType(SetType("Y")))]
type_with_known_types(root, env, lst, ["r","x"])
assert isinstance(get_type_by_name(env, "x"), SetType)
assert get_type_by_name(env, "x").name == "Y"
def test_types_relation_set_enum(self):
# Build AST
string_to_file("#PREDICATE r = {8|->10, 7|->11, 2|->11, 6|->12}", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["r"])
assert isinstance(get_type_by_name(env, "r"), PowerSetType)
assert isinstance(get_type_by_name(env, "r").data, CartType)
assert isinstance(get_type_by_name(env, "r").data.left.data, IntegerType)
assert isinstance(get_type_by_name(env, "r").data.right.data, IntegerType)
def test_types_ran_unify(self):
# Build AST
string_to_file("#PREDICATE d=A*B & c=ran(d) & A=NAT & B=NAT", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
env = Environment()
type_with_known_types(root, env, [], ["c","d","A","B"])
assert isinstance(get_type_by_name(env, "A"), PowerSetType)
assert isinstance(get_type_by_name(env, "B"), PowerSetType)
assert isinstance(get_type_by_name(env, "c"), PowerSetType)
assert isinstance(get_type_by_name(env, "d"), PowerSetType)
assert isinstance(get_type_by_name(env, "d").data, CartType)
def test_types_seq_extention(self):
# Build AST
string_to_file("#PREDICATE s=[1,2,3]", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["s"])
assert isinstance(get_type_by_name(env, "s"), PowerSetType)
assert isinstance(get_type_by_name(env, "s").data, CartType)
assert isinstance(get_type_by_name(env, "s").data.left.data, IntegerType)
assert isinstance(get_type_by_name(env, "s").data.right.data, IntegerType)
def test_types_set_cart_unify4(self):
# Build AST
string_to_file("#PREDICATE NAT*B=A*POW(NAT)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["A","B"])
assert isinstance(get_type_by_name(env, "A"), PowerSetType)
assert isinstance(get_type_by_name(env, "B"), PowerSetType)
assert isinstance(get_type_by_name(env, "A").data, IntegerType)
assert isinstance(get_type_by_name(env, "B").data, PowerSetType)
assert isinstance(get_type_by_name(env, "B").data.data, IntegerType)
def test_types_set_cart_tuple2(self):
# Build AST:
string_to_file("#PREDICATE x=(1,2,41) & y={(0,0,41),(1,0,41),(0,1,41),(1,1,41)}", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
type_with_known_types(root, env, [], ["x","y"])
assert isinstance(get_type_by_name(env, "x"), CartType)
assert isinstance(get_type_by_name(env, "y"), PowerSetType)
assert isinstance(get_type_by_name(env, "y").data, CartType)
assert isinstance(get_type_by_name(env, "y").data.left.data, CartType)
assert isinstance(get_type_by_name(env, "y").data.right.data, IntegerType)
def test_types_seq_conc(self):
# Build AST
string_to_file("#PREDICATE ss:perm(perm(S)) & s=conc(ss)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X")))]
type_with_known_types(root, env, lst, ["s", "ss"])
assert isinstance(get_type_by_name(env, "s"), PowerSetType)
assert isinstance(get_type_by_name(env, "s").data, CartType)
assert isinstance(get_type_by_name(env, "ss"), PowerSetType)
assert isinstance(get_type_by_name(env, "ss").data, CartType)
def test_types_simple_parprod(self):
# Build AST
string_to_file("#PREDICATE r1:A<->B & r2:C<->D & r3=(r1 || r2)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("A", PowerSetType(SetType("X"))),("B", PowerSetType(SetType("Y"))),("C", PowerSetType(SetType("M"))),("D", PowerSetType(SetType("N")))]
type_with_known_types(root, env, lst, ["r1","r2","r3"])
assert isinstance(get_type_by_name(env, "r3"), PowerSetType)
assert isinstance(get_type_by_name(env, "r3").data, CartType)
assert isinstance(get_type_by_name(env, "r3").data.left.data, CartType)
assert isinstance(get_type_by_name(env, "r3").data.right.data, CartType)
x = get_type_by_name(env, "r3").data.left.data.left.data
y = get_type_by_name(env, "r3").data.left.data.right.data
m = get_type_by_name(env, "r3").data.right.data.left.data
n = get_type_by_name(env, "r3").data.right.data.right.data
assert isinstance(x, SetType)
assert isinstance(m, SetType)
assert isinstance(y, SetType)
assert isinstance(n, SetType)
assert x.name == "X"
assert y.name == "M"
assert m.name == "Y"
assert n.name == "N"
def test_types_simple_closure1(self):
# Build AST
string_to_file("#PREDICATE f=closure1(r)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
l = [("r", PowerSetType(CartType(PowerSetType(IntegerType()),PowerSetType(IntegerType()))))]
type_with_known_types(root, env, l, ["f"])
assert isinstance(get_type_by_name(env, "f"), PowerSetType)
assert isinstance(get_type_by_name(env, "f").data, CartType)
assert isinstance(get_type_by_name(env, "r"), PowerSetType)
assert isinstance(get_type_by_name(env, "r").data, CartType)
def test_types_seq_append(self):
# Build AST
string_to_file("#PREDICATE s:perm(S) & E:S & t=s<-E", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X")))]
type_with_known_types(root, env, lst, ["s", "t", "E"])
assert isinstance(get_type_by_name(env, "t"), PowerSetType)
assert isinstance(get_type_by_name(env, "t").data, CartType)
assert isinstance(get_type_by_name(env, "t").data.left.data, IntegerType)
assert isinstance(get_type_by_name(env, "t").data.right.data, SetType)
def test_types_rel_repr(self):
# Build AST:
string_to_file("#PREDICATE f={aa|->aa, aa|->bb, bb|->bb, bb|->aa} & g=A*A", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("A", PowerSetType(SetType("X"))),("aa",SetType("X")),("bb",SetType("X"))]
type_with_known_types(root, env, lst, ["f","g"])
assert isinstance(get_type_by_name(env, "g"), PowerSetType)
assert isinstance(get_type_by_name(env, "g").data, CartType)
assert isinstance(get_type_by_name(env, "f"), PowerSetType)
assert isinstance(get_type_by_name(env, "f").data, CartType)
def test_types_set_gen_inter(self):
# Build AST:
string_to_file("#PREDICATE U:POW(POW(S)) & u=inter(U)", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("S")))]
type_with_known_types(root, env, lst, ["U","u"])
assert isinstance(get_type_by_name(env, "U"), PowerSetType)
assert isinstance(get_type_by_name(env, "U").data, PowerSetType)
assert isinstance(get_type_by_name(env, "U").data.data, SetType)
assert isinstance(get_type_by_name(env, "u"), PowerSetType)
assert isinstance(get_type_by_name(env, "u").data, SetType)
def test_types_simple_set_com2(self):
# Build AST
string_to_file("#PREDICATE S={a,b} & ID={x,y|x<10 & y:S}", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("a", SetType("X")),("b", SetType("X"))]
type_with_known_types(root, env, lst, ["S","ID"])
assert isinstance(get_type_by_name(env, "S"), PowerSetType)
assert isinstance(get_type_by_name(env, "ID"), PowerSetType)
assert isinstance(get_type_by_name(env, "x"), IntegerType)
assert isinstance(get_type_by_name(env, "y"), SetType)
assert get_type_by_name(env, "y").name =="X"
def test_types_relation(self):
# Build AST
string_to_file("#PREDICATE r=S<->T", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
lst = [("S", PowerSetType(SetType("X"))),("T", PowerSetType(SetType("X")))]
type_with_known_types(root, env, lst, ["r"])
assert isinstance(get_type_by_name(env, "r"), PowerSetType)
assert isinstance(get_type_by_name(env, "r").data, PowerSetType)
assert isinstance(get_type_by_name(env, "r").data.data, CartType)
assert isinstance(get_type_by_name(env, "r").data.data.left.data, SetType)
assert isinstance(get_type_by_name(env, "r").data.data.right.data, SetType)
def test_types_simple_set(self):
# Build AST
string_to_file("#PREDICATE ID={aa,bb}", file_name)
ast_string = file_to_AST_str(file_name)
root = str_ast_to_python_ast(ast_string)
# Type
env = Environment()
env.add_ids_to_frame(["aa","bb","ID"])
env.set_value("aa", "aa")
env.set_value("bb", "bb")
env.set_value("ID", frozenset(["aa", "bb"]))
lst = [("aa", SetType("X")),("bb", SetType("X"))]
type_with_known_types(root, env, lst, ["ID"])
assert isinstance(get_type_by_name(env, "ID"), PowerSetType)
assert isinstance(get_type_by_name(env, "ID").data, SetType)
assert get_type_by_name(env, "ID").data.name =="X"
