Fix tests and add one

This commit is contained in:
Scott Shawcroft 2021-12-22 12:35:25 -08:00
parent 210873997e
commit 6fd0518867
No known key found for this signature in database
GPG key ID: 0DFD512649C052DA
2 changed files with 8 additions and 0 deletions

View file

@ -38,6 +38,7 @@ def assemble(text_program):
labels = {} labels = {}
instructions = [] instructions = []
sideset_count = 0 sideset_count = 0
sideset_enable = 0
for line in text_program.split("\n"): for line in text_program.split("\n"):
line = line.strip() line = line.strip()
if not line: if not line:

View file

@ -44,6 +44,13 @@ class TestNop(unittest.TestCase):
self.assertAssemblesTo(".side_set 1\nnop side 1", [0b101_10000_010_00_010]) self.assertAssemblesTo(".side_set 1\nnop side 1", [0b101_10000_010_00_010])
self.assertAssemblesTo(".side_set 1\nnop side 1 [1]", [0b101_10001_010_00_010]) self.assertAssemblesTo(".side_set 1\nnop side 1 [1]", [0b101_10001_010_00_010])
def testSidesetOpt(self):
self.assertAssemblesTo(".side_set 1 opt\nnop side 1", [0b101_11000_010_00_010])
self.assertAssemblesTo(
".side_set 1 opt\nnop side 0 [1]", [0b101_10001_010_00_010]
)
self.assertAssemblesTo(".side_set 1 opt\nnop [1]", [0b101_00001_010_00_010])
def testJmp(self): def testJmp(self):
self.assertAssemblesTo("l:\njmp l", [0b000_00000_000_00000]) self.assertAssemblesTo("l:\njmp l", [0b000_00000_000_00000])
self.assertAssemblesTo("l:\njmp 7", [0b000_00000_000_00111]) self.assertAssemblesTo("l:\njmp 7", [0b000_00000_000_00111])