69 lines
1.3 KiB
Python
69 lines
1.3 KiB
Python
from io import StringIO
|
|
import pytest
|
|
|
|
from assembler.assembler import Assembler
|
|
|
|
|
|
def test_commands(basic_machine_definition):
|
|
memory_definition, command_defintion = basic_machine_definition
|
|
|
|
data = StringIO(
|
|
'''
|
|
ldi r0, 0xfe
|
|
ldi r1, 0xfe
|
|
add r0, r1
|
|
'''
|
|
)
|
|
assembler = Assembler(data, memory_definition, command_defintion, {})
|
|
assembler.parse()
|
|
|
|
assert assembler._code_objects == [32704 | 0, 0xfe
|
|
, 32704 | 1, 0xfe
|
|
, 40896 | 0, 1]
|
|
|
|
def test_mark(basic_machine_definition):
|
|
memory_definition, command_defintion = basic_machine_definition
|
|
|
|
data = StringIO(
|
|
'''
|
|
ldi r0, test_mark
|
|
ldi r1, 0xfe
|
|
test_mark:
|
|
add r0, r1
|
|
'''
|
|
)
|
|
assembler = Assembler(data, memory_definition, command_defintion, {})
|
|
assembler.parse()
|
|
|
|
|
|
assert assembler._code_objects == [32704 | 0, 4
|
|
, 32704 | 1, 0xfe
|
|
, 40896 | 0, 1]
|
|
|
|
|
|
def test_set_directive(basic_machine_definition):
|
|
memory_definition, command_defintion = basic_machine_definition
|
|
|
|
data = StringIO(
|
|
'''
|
|
ldi r0, test_mark
|
|
ldi r1, 0xfe
|
|
test_mark:
|
|
.set [0xfe, 0xef,
|
|
10, 20,
|
|
'a', 'b',
|
|
'\\n', 0b10]
|
|
'''
|
|
)
|
|
assembler = Assembler(data, memory_definition, command_defintion, {})
|
|
assembler.parse()
|
|
|
|
|
|
assert assembler._code_objects == [32704 | 0, 4
|
|
, 32704 | 1, 0xfe
|
|
, 0xfe, 0xef
|
|
, 10, 20
|
|
, ord("a"), ord("b")
|
|
, ord("\n"), 0b10]
|
|
|