test_instruction_break.py 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. import unittest
  2. from modelverse_kernel.main import ModelverseKernel
  3. from utils import execute_until_finished, MvSWrapper, get_inst, get_phase, get_returnvalue
  4. class TestBreak(unittest.TestCase):
  5. def setUp(self):
  6. self.mvs = MvSWrapper()
  7. self.root = self.mvs.execute("RR", [])[0]
  8. self.mvk = ModelverseKernel(self.root)
  9. self.user_root = self.mvs.execute("CN", [])[0]
  10. self.user_frame = self.mvs.execute("CN", [])[0]
  11. self.symbols = self.mvs.execute("CN", [])[0]
  12. self.evalstack = self.mvs.execute("CN", [])[0]
  13. self.returnvalue = self.mvs.execute("CN", [])[0]
  14. self.phase = self.mvs.execute("CNV", ["init"])[0]
  15. self.mvs.execute("CD", [self.root, "user_1", self.user_root])
  16. self.mvs.execute("CD", [self.user_root, "frame", self.user_frame])
  17. self.mvs.execute("CD", [self.user_frame, "symbols", self.symbols])
  18. self.mvs.execute("CD", [self.user_frame, "evalstack", self.evalstack])
  19. self.mvs.execute("CD", [self.user_frame, "phase", self.phase])
  20. self.mvs.execute("CD", [self.user_frame, "returnvalue", self.returnvalue])
  21. def test_break(self):
  22. """
  23. Test the code:
  24. while True:
  25. a
  26. break
  27. b
  28. c
  29. """
  30. self.inst = self.mvs.execute("CNV", [{"value":"while"}])[0]
  31. self.const_a = self.mvs.execute("CNV", [{"value":"constant"}])[0]
  32. self.const_b = self.mvs.execute("CNV", [{"value":"constant"}])[0]
  33. self.const_c = self.mvs.execute("CNV", [{"value":"constant"}])[0]
  34. self.const_cond = self.mvs.execute("CNV", [{"value":"constant"}])[0]
  35. self.value_a = self.mvs.execute("CNV", ["a"])[0]
  36. self.value_b = self.mvs.execute("CNV", ["b"])[0]
  37. self.value_c = self.mvs.execute("CNV", ["c"])[0]
  38. self.value_cond = self.mvs.execute("CNV", [True])[0]
  39. self.break_inst = self.mvs.execute("CNV", [{"value":"break"}])[0]
  40. self.mvs.execute("CD", [self.user_frame, "IP", self.inst])
  41. self.mvs.execute("CD", [self.inst, "cond", self.const_cond])
  42. self.mvs.execute("CD", [self.inst, "body", self.const_a])
  43. self.mvs.execute("CD", [self.inst, "next", self.const_c])
  44. self.mvs.execute("CD", [self.const_a, "node", self.value_a])
  45. self.mvs.execute("CD", [self.const_a, "next", self.break_inst])
  46. self.mvs.execute("CD", [self.break_inst, "next", self.const_b])
  47. self.mvs.execute("CD", [self.break_inst, "while", self.inst])
  48. self.mvs.execute("CD", [self.const_b, "node", self.value_b])
  49. self.mvs.execute("CD", [self.const_c, "node", self.value_c])
  50. self.mvs.execute("CD", [self.const_cond, "node", self.value_cond])
  51. ### Execute rules
  52. self.assertEquals(get_inst(self.root, self.mvs), "while")
  53. self.assertEquals(get_phase(self.root, self.mvs), "init")
  54. execute_until_finished(self.mvk, self.mvs)
  55. self.assertEquals(get_inst(self.root, self.mvs), "constant")
  56. self.assertEquals(get_phase(self.root, self.mvs), "init")
  57. execute_until_finished(self.mvk, self.mvs)
  58. self.assertEquals(get_inst(self.root, self.mvs), "constant")
  59. self.assertEquals(get_phase(self.root, self.mvs), "finish")
  60. self.assertEquals(get_returnvalue(self.root, self.mvs), True)
  61. execute_until_finished(self.mvk, self.mvs)
  62. self.assertEquals(get_inst(self.root, self.mvs), "while")
  63. self.assertEquals(get_phase(self.root, self.mvs), "cond")
  64. execute_until_finished(self.mvk, self.mvs)
  65. self.assertEquals(get_inst(self.root, self.mvs), "constant")
  66. self.assertEquals(get_phase(self.root, self.mvs), "init")
  67. execute_until_finished(self.mvk, self.mvs)
  68. self.assertEquals(get_inst(self.root, self.mvs), "constant")
  69. self.assertEquals(get_phase(self.root, self.mvs), "finish")
  70. self.assertEquals(get_returnvalue(self.root, self.mvs), "a")
  71. execute_until_finished(self.mvk, self.mvs)
  72. self.assertEquals(get_inst(self.root, self.mvs), "break")
  73. self.assertEquals(get_phase(self.root, self.mvs), "init")
  74. execute_until_finished(self.mvk, self.mvs)
  75. self.assertEquals(get_inst(self.root, self.mvs), "while")
  76. self.assertEquals(get_phase(self.root, self.mvs), "finish")
  77. execute_until_finished(self.mvk, self.mvs)
  78. self.assertEquals(get_inst(self.root, self.mvs), "constant")
  79. self.assertEquals(get_phase(self.root, self.mvs), "init")
  80. execute_until_finished(self.mvk, self.mvs)
  81. self.assertEquals(get_inst(self.root, self.mvs), "constant")
  82. self.assertEquals(get_phase(self.root, self.mvs), "finish")
  83. self.assertEquals(get_returnvalue(self.root, self.mvs), "c")