import unittest from modelverse_kernel.main import ModelverseKernel from utils import execute_until_finished, MvSWrapper, get_inst, get_phase, get_returnvalue, get_phase_ref, read_primitive_interfaces, get_returnvalue_ref class TestSet(unittest.TestCase): def setUp(self): self.mvs = MvSWrapper() self.root = self.mvs.execute("RR", [])[0] self.mvk = ModelverseKernel(self.root) self.task_root = self.mvs.execute("CN", [])[0] self.task_frame = self.mvs.execute("CN", [])[0] self.symbols = self.mvs.execute("CN", [])[0] self.evalstack = self.mvs.execute("CN", [])[0] self.returnvalue = self.mvs.execute("CN", [])[0] self.phase = self.mvs.execute("CNV", ["init"])[0] self.mvs.execute("CD", [self.root, "task_1", self.task_root]) self.mvs.execute("CD", [self.task_root, "frame", self.task_frame]) self.mvs.execute("CD", [self.task_frame, "symbols", self.symbols]) self.mvs.execute("CD", [self.task_frame, "evalstack", self.evalstack]) self.mvs.execute("CD", [self.task_frame, "phase", self.phase]) self.mvs.execute("CD", [self.task_frame, "returnvalue", self.returnvalue]) execute_until_finished(self.mvk, self.mvs, "load_primitives", []) self.primitives = read_primitive_interfaces(self.root, self.mvs) def test_add(self): self.helper_primitives_2_params("set_add", set([1, 2, 3]), 4, set([1, 2, 3, 4])) def test_add_empty(self): self.helper_primitives_2_params("set_add", set([]), "abc", set(["abc"])) def test_pop(self): self.helper_primitives_1_params("set_pop", set([1]), 1) def test_remove(self): self.helper_primitives_2_params("set_remove", set([1, 2, 3, 4]), 3, set([1, 2, 4])) def test_remove_last(self): self.helper_primitives_2_params("set_remove", set([1]), 1, set([])) def test_in_T(self): self.helper_primitives_2_params("set_in", set([1]), 1, True) def test_in_T_multi(self): self.helper_primitives_2_params("set_in", set([1, 2, 3]), 1, True) def test_in_F(self): self.helper_primitives_2_params("set_in", set([1]), 2, False) def test_in_empty(self): self.helper_primitives_2_params("set_in", set([]), 2, False) def construct(self, source, lst): for v in lst: t = self.mvs.execute("CNV", [v])[0] self.mvs.execute("CE", [source, t]) def parse(self, source, expected): if isinstance(expected, set): self.assertEquals(len(self.mvs.execute("RO", [source])[0]), len(expected)) s = set() for v in self.mvs.execute("RO", [source])[0]: s.add(self.mvs.execute("RV", [self.mvs.execute("RE", [v])[0][1]])[0]) self.assertEquals(s, expected) else: self.assertEquals(get_returnvalue(self.root, self.mvs), expected) def helper_primitives_2_params(self, operation, a, b, result): self.actual_arg_a = self.mvs.execute("CN", [])[0] self.actual_arg_b = self.mvs.execute("CN", [])[0] self.inst = self.mvs.execute("CNV", [{"value":"call"}])[0] self.const_a = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_b = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_c = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_first = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_second = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.value_a = self.mvs.execute("CNV", ["a"])[0] self.value_b = self.mvs.execute("CNV", ["b"])[0] self.value_c = self.mvs.execute("CNV", ["c"])[0] self.value_first = self.mvs.execute("CN", [])[0] self.value_second = self.mvs.execute("CNV", [b])[0] self.construct(self.value_first, a) self.mvs.execute("CD", [self.task_frame, "IP", self.inst]) self.const_func = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.mvs.execute("CD", [self.inst, "func", self.const_func]) self.mvs.execute("CD", [self.const_func, "node", self.primitives[operation]]) self.mvs.execute("CD", [self.inst, "next", self.const_c]) self.mvs.execute("CD", [self.inst, "params", self.actual_arg_a]) self.mvs.execute("CD", [self.actual_arg_a, "next_param", self.actual_arg_b]) self.mvs.execute("CD", [self.inst, "last_param", self.actual_arg_b]) self.mvs.execute("CD", [self.actual_arg_a, "value", self.const_first]) self.mvs.execute("CD", [self.actual_arg_b, "value", self.const_second]) self.mvs.execute("CD", [self.actual_arg_a, "name", self.value_a]) self.mvs.execute("CD", [self.actual_arg_b, "name", self.value_b]) self.mvs.execute("CD", [self.const_a, "node", self.value_a]) self.mvs.execute("CD", [self.const_b, "node", self.value_b]) self.mvs.execute("CD", [self.const_c, "node", self.value_c]) self.mvs.execute("CD", [self.const_first, "node", self.value_first]) self.mvs.execute("CD", [self.const_second, "node", self.value_second]) ### Execute rules self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase_ref(self.root, self.mvs), self.actual_arg_a) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.assertEquals(get_returnvalue_ref(self.root, self.mvs), self.value_first) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase_ref(self.root, self.mvs), self.actual_arg_b) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.assertEquals(get_returnvalue(self.root, self.mvs), b) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "call") execute_until_finished(self.mvk, self.mvs) # Body points to empty instruction, as it is a primitive self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) # This will have triggered the primitive, so now we are back where we left off self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "finish") execute_until_finished(self.mvk, self.mvs) self.parse(self.value_first, result) def helper_primitives_1_params(self, operation, a, result): self.actual_arg_a = self.mvs.execute("CN", [])[0] self.inst = self.mvs.execute("CNV", [{"value":"call"}])[0] self.const_a = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_b = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_c = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_first = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.value_a = self.mvs.execute("CNV", ["a"])[0] self.value_b = self.mvs.execute("CNV", ["b"])[0] self.value_c = self.mvs.execute("CNV", ["c"])[0] self.value_first = self.mvs.execute("CN", [])[0] self.mvs.execute("CD", [self.task_frame, "IP", self.inst]) self.const_func = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.mvs.execute("CD", [self.inst, "func", self.const_func]) self.mvs.execute("CD", [self.const_func, "node", self.primitives[operation]]) self.mvs.execute("CD", [self.inst, "next", self.const_c]) self.mvs.execute("CD", [self.inst, "params", self.actual_arg_a]) self.mvs.execute("CD", [self.inst, "last_param", self.actual_arg_a]) self.mvs.execute("CD", [self.actual_arg_a, "value", self.const_first]) self.mvs.execute("CD", [self.actual_arg_a, "name", self.value_a]) self.mvs.execute("CD", [self.const_a, "node", self.value_a]) self.mvs.execute("CD", [self.const_b, "node", self.value_b]) self.mvs.execute("CD", [self.const_c, "node", self.value_c]) self.mvs.execute("CD", [self.const_first, "node", self.value_first]) self.construct(self.value_first, a) ### Execute rules self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase_ref(self.root, self.mvs), self.actual_arg_a) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.assertEquals(get_returnvalue_ref(self.root, self.mvs), self.value_first) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "call") execute_until_finished(self.mvk, self.mvs) # Body points to empty instruction, as it is a primitive self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) # This will have triggered the primitive, so now we are back where we left off # Just check whether or not the correct value is in the returnvalue to be used self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.parse(self.value_first, result) def helper_primitives_3_params(self, operation, a, b, c, result): self.actual_arg_a = self.mvs.execute("CN", [])[0] self.actual_arg_b = self.mvs.execute("CN", [])[0] self.actual_arg_c = self.mvs.execute("CN", [])[0] self.inst = self.mvs.execute("CNV", [{"value":"call"}])[0] self.const_a = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_b = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_c = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_first = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_second = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.const_third = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.value_a = self.mvs.execute("CNV", ["a"])[0] self.value_b = self.mvs.execute("CNV", ["b"])[0] self.value_c = self.mvs.execute("CNV", ["c"])[0] self.value_first = self.mvs.execute("CN", [])[0] self.value_second = self.mvs.execute("CNV", [b])[0] self.value_third = self.mvs.execute("CNV", [c])[0] self.mvs.execute("CD", [self.task_frame, "IP", self.inst]) self.const_func = self.mvs.execute("CNV", [{"value":"constant"}])[0] self.mvs.execute("CD", [self.inst, "func", self.const_func]) self.mvs.execute("CD", [self.const_func, "node", self.primitives[operation]]) self.mvs.execute("CD", [self.inst, "next", self.const_c]) self.mvs.execute("CD", [self.inst, "params", self.actual_arg_a]) self.mvs.execute("CD", [self.actual_arg_a, "next_param", self.actual_arg_b]) self.mvs.execute("CD", [self.actual_arg_b, "next_param", self.actual_arg_c]) self.mvs.execute("CD", [self.inst, "last_param", self.actual_arg_c]) self.mvs.execute("CD", [self.actual_arg_a, "value", self.const_first]) self.mvs.execute("CD", [self.actual_arg_b, "value", self.const_second]) self.mvs.execute("CD", [self.actual_arg_c, "value", self.const_third]) self.mvs.execute("CD", [self.actual_arg_a, "name", self.value_a]) self.mvs.execute("CD", [self.actual_arg_b, "name", self.value_b]) self.mvs.execute("CD", [self.actual_arg_c, "name", self.value_c]) self.mvs.execute("CD", [self.const_a, "node", self.value_a]) self.mvs.execute("CD", [self.const_b, "node", self.value_b]) self.mvs.execute("CD", [self.const_c, "node", self.value_c]) self.mvs.execute("CD", [self.const_first, "node", self.value_first]) self.mvs.execute("CD", [self.const_second, "node", self.value_second]) self.mvs.execute("CD", [self.const_third, "node", self.value_third]) self.construct(self.value_first, a) ### Execute rules self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase_ref(self.root, self.mvs), self.actual_arg_a) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.assertEquals(get_returnvalue_ref(self.root, self.mvs), self.value_first) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase_ref(self.root, self.mvs), self.actual_arg_b) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.assertEquals(get_returnvalue(self.root, self.mvs), b) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase_ref(self.root, self.mvs), self.actual_arg_c) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "constant") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.assertEquals(get_returnvalue(self.root, self.mvs), c) execute_until_finished(self.mvk, self.mvs) self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "call") execute_until_finished(self.mvk, self.mvs) # Body points to empty instruction, as it is a primitive self.assertEquals(get_phase(self.root, self.mvs), "init") execute_until_finished(self.mvk, self.mvs) # This will have triggered the primitive, so now we are back where we left off # Just check whether or not the correct value is in the returnvalue to be used self.assertEquals(get_inst(self.root, self.mvs), "call") self.assertEquals(get_phase(self.root, self.mvs), "finish") self.parse(self.value_first, result)