github.com/rzzzwilson.pymlac
1
0
Fork 0
mirror of https://github.com/rzzzwilson/pymlac.git synced 2025-06-10 09:32:41 +00:00
Code Issues Releases Wiki Activity

More tests

Browse Source
This commit is contained in:
Ross Wilson
2016-01-26 13:52:16 +07:00
parent 5700127b71
commit 61122bb354
1 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
pyasm/test_pyasm.py
View File

@@ -84,6 +84,31 @@ class TestPyasm(unittest.TestCase):
% (test, str(expected), str(result)))
self.assertEqual(result, expected, msg)
def test_define_label(self):
"""Run lots of tests on define_label().
Note that we CAN'T test defining a label twice as define_line() will
call sys.exit().
"""
# define tests: (label, address, lnum, expected_symtab, expected_symtabline
tests = (
('A', 1, 2, {'A': 1}, {'A': 2}),
('B', 0100, 5, {'B': 0100}, {'B': 5}),
)
# now run them
for (label, address, lnum, exp_symtab, exp_symtabline) in tests:
pyasm.SymTable = {}
pyasm.SymTableLine = {}
pyasm.define_label(label, address, lnum)
msg = ("Expected define_label('%s', %d, %d) to set SymTable '%s', got '%s'"
% (label, address, lnum, str(exp_symtab), str(pyasm.SymTable)))
self.assertEqual(exp_symtab, pyasm.SymTable, msg)
msg = ("Expected define_label('%s', %d, %d) to set SymTableLine '%s', got '%s'"
% (label, address, lnum, str(exp_symtabline), str(pyasm.SymTableLine)))
self.assertEqual(exp_symtabline, pyasm.SymTableLine, msg)
################################################################################
if __name__ == '__main__':