mirror of
https://github.com/rzzzwilson/pymlac.git
synced 2025-06-10 09:32:41 +00:00
118 lines
4.9 KiB
Python
118 lines
4.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Test the pyasm functions.
|
|
|
|
Usage: test_pyasm.py [<options>]
|
|
|
|
where <options> is one or more of
|
|
-h prints this help and stops
|
|
"""
|
|
|
|
import os
|
|
import pyasm
|
|
import unittest
|
|
|
|
|
|
class TestPyasm(unittest.TestCase):
|
|
|
|
def test_eval_expression(self):
|
|
"""Run tests on eval(expression).
|
|
|
|
eval_expression(expr, dot) with SymTable global
|
|
"""
|
|
|
|
# define tests: (test, dot, symtable, expected, undefined, raises)
|
|
tests = (
|
|
('a', 0100, {'A': 1}, 1, None, False),
|
|
('.', 0100, {}, 0100, None, False),
|
|
('b', 0100, {}, None, 'B', True),
|
|
('a+1', 0100, {'A': 1}, 2, None, False),
|
|
('a+b', 0, {'A': 1, 'B': 2}, 3, None, False),
|
|
('a + b', 0, {'A': 1, 'B': 2}, 3, None, False),
|
|
('a / b', 0, {'A': 4, 'B': 2}, 2, None, False),
|
|
('. + 0100', 0100, {}, 0200, None, False),
|
|
)
|
|
|
|
# now run them
|
|
for (test, dot, symtable, expected, undefined, raises) in tests:
|
|
pyasm.Undefined = None
|
|
pyasm.SymTable = symtable
|
|
if raises:
|
|
result = None
|
|
with self.assertRaises(NameError):
|
|
result = pyasm.eval_expression(test, dot)
|
|
else:
|
|
result = pyasm.eval_expression(test, dot)
|
|
msg = ("Expected eval_expression('%s', '%s') to return '%s', got '%s'"
|
|
% (test, str(dot), str(expected), str(result)))
|
|
self.assertEqual(result, expected, msg)
|
|
msg = ("Expected eval_expression('%s', '%s') to set Undefined to '%s', got '%s'"
|
|
% (test, str(dot), str(undefined), str(pyasm.Undefined)))
|
|
self.assertEqual(undefined, pyasm.Undefined, msg)
|
|
|
|
def test_split_fields(self):
|
|
"""Run lots of tests on split_fields()."""
|
|
|
|
# define tests: (test, expected)
|
|
tests = (('label opcode value ;comment', ('LABEL', 'OPCODE', 'value')),
|
|
('', (None, None, None)),
|
|
(';comment', (None, None, None)),
|
|
('; comment', (None, None, None)),
|
|
(' ; comment', (None, None, None)),
|
|
('\t; comment', (None, None, None)),
|
|
('label', ('LABEL', None, None)),
|
|
(' opcode', (None, 'OPCODE', None)),
|
|
('\topcode', (None, 'OPCODE', None)),
|
|
('label opcode', ('LABEL', 'OPCODE', None)),
|
|
('label\topcode', ('LABEL', 'OPCODE', None)),
|
|
('label opcode value', ('LABEL', 'OPCODE', 'value')),
|
|
('label\topcode value', ('LABEL', 'OPCODE', 'value')),
|
|
('label\topcode\tvalue', ('LABEL', 'OPCODE', 'value')),
|
|
('label opcode a+b', ('LABEL', 'OPCODE', 'a+b')),
|
|
('label opcode a + b', ('LABEL', 'OPCODE', 'a + b')),
|
|
(' opcode a + b', (None, 'OPCODE', 'a + b')),
|
|
(' opcode a + b ;comment', (None, 'OPCODE', 'a + b')),
|
|
(' opcode', (None, 'OPCODE', None)),
|
|
)
|
|
|
|
# now run them
|
|
for (test, expected) in tests:
|
|
result = pyasm.split_fields(test)
|
|
msg = ("Expected split_fields('%s') to return '%s', got '%s'"
|
|
% (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__':
|
|
suite = unittest.makeSuite(TestPyasm, 'test')
|
|
runner = unittest.TextTestRunner()
|
|
runner.run(suite)
|