test_optimize.py 496 B

12345678910111213141516171819
  1. import unittest
  2. from src.optimize import optimize_global
  3. from src.statement import Statement as S, Block as B
  4. class TestOptimize(unittest.TestCase):
  5. def setUp(self):
  6. pass
  7. def test_optimize_global_movaa(self):
  8. foo = S('command', 'foo')
  9. bar = S('command', 'baz')
  10. block = B([foo, \
  11. S('command', 'move', '$regA', '$regA'),
  12. bar])
  13. optimize_global(block)
  14. self.assertEquals(block.statements, [foo, bar])