123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166 |
- 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 TestSetNode(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_remove(self):
- n = self.mvs.execute("CN", [])[0]
- a = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, a])
- b = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, b])
- c = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, c])
- self.helper_primitives_2_params("set_remove_node", n, b, set([a, c]))
- def test_remove_last(self):
- n = self.mvs.execute("CN", [])[0]
- a = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, a])
- self.helper_primitives_2_params("set_remove_node", n, a, set([]))
- def test_in_T(self):
- n = self.mvs.execute("CN", [])[0]
- a = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, a])
- b = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, b])
- c = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, c])
- self.helper_primitives_2_params("set_in_node", n, c, True)
- def test_in_F(self):
- n = self.mvs.execute("CN", [])[0]
- a = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, a])
- b = self.mvs.execute("CN", [])[0]
- self.mvs.execute("CE", [n, b])
- c = self.mvs.execute("CN", [])[0]
- self.helper_primitives_2_params("set_in_node", n, c, False)
- def test_in_empty(self):
- n = self.mvs.execute("CN", [])[0]
- c = self.mvs.execute("CN", [])[0]
- self.helper_primitives_2_params("set_in_node", n, c, False)
- 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("RE", [v])[0][1])
- 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 = b
- 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")
- 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)
|