| 1234567891011121314151617181920212223242526 |
- from unittest import TestCase
- from src.parser import Parser
- from tests.parser import ParserWrapper
- def rewrite(exp, **kwargs):
- return ParserWrapper(Parser, **kwargs).run([exp, '@'])
- class TestRewrite(TestCase):
- def assertRewrite(self, rewrite_chain):
- try:
- for i, exp in enumerate(rewrite_chain[:-1]):
- self.assertEqual(str(rewrite(exp)), str(rewrite_chain[i+1]))
- except AssertionError: # pragma: nocover
- print 'rewrite failed:', exp, '->', rewrite_chain[i+1]
- print 'rewrite chain:', rewrite_chain
- raise
- def test_addition_rewrite(self):
- self.assertRewrite(['2 + 3 + 4', '5 + 4', '9'])
- def test_addition_identifiers_rewrite(self):
- self.assertRewrite(['2 + 3a + 4', '6 + 3a'])
|
