| import segprint2fasm |
| |
| import unittest |
| import StringIO |
| import re |
| |
| |
| class TestStringMethods(unittest.TestCase): |
| def check_segprint_fasm_equiv(self, segp_fn, fasm_fn): |
| fout = StringIO.StringIO() |
| segprint2fasm.run(open(segp_fn, 'r'), fout) |
| fasm_out = fout.getvalue() |
| |
| fasm_ref = open(fasm_fn, 'r').read() |
| |
| def normalize(fasm): |
| '''Remove all comments and sort''' |
| ret = [] |
| for l in fasm.split('\n'): |
| # Remove comments |
| i = l.rfind('#') |
| if i >= 0: |
| l = l[0:i] |
| l = l.strip() |
| if not l: |
| continue |
| ret.append(l) |
| return sorted(ret) |
| |
| fasm_out = normalize(fasm_out) |
| fasm_ref = normalize(fasm_ref) |
| self.assertEquals(fasm_ref, fasm_out) |
| |
| def test_lut_int(self): |
| self.check_segprint_fasm_equiv( |
| 'test_data/lut_int/design.segp', 'test_data/lut_int.fasm') |
| |
| def test_ff_int(self): |
| self.check_segprint_fasm_equiv( |
| 'test_data/ff_int/design.segp', 'test_data/ff_int.fasm') |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
