prog8/tests/test_zp.py
Irmen de Jong 12c3ddd207 renames
2018-01-07 23:48:31 +01:00

60 lines
1.9 KiB
Python

import pytest
from il65.handwritten.symbols import Zeropage, SymbolError, DataType
def test_zp_configure_onlyonce():
zp = Zeropage()
zp.configure()
with pytest.raises(SymbolError):
zp.configure()
def test_zp_names():
zp = Zeropage()
zp.configure()
zp.allocate("", DataType.BYTE)
zp.allocate("", DataType.BYTE)
zp.allocate("varname", DataType.BYTE)
with pytest.raises(AssertionError):
zp.allocate("varname", DataType.BYTE)
zp.allocate("varname2", DataType.BYTE)
def test_zp_noclobber_allocation():
zp = Zeropage()
zp.configure(False)
assert zp.available() == 9
with pytest.raises(LookupError):
zp.allocate("impossible", DataType.FLOAT) # in regular zp there aren't 5 sequential bytes free
for i in range(zp.available()):
zp.allocate("bytevar"+str(i), DataType.BYTE)
assert zp.available() == 0
with pytest.raises(LookupError):
zp.allocate("", DataType.BYTE)
with pytest.raises(LookupError):
zp.allocate("", DataType.WORD)
def test_zp_clobber_allocation():
zp = Zeropage()
zp.configure(True)
assert zp.available() == 239
loc = zp.allocate("", DataType.FLOAT)
assert loc > 3 and loc not in zp.free
num, rest = divmod(zp.available(), 5)
for _ in range(num-3):
zp.allocate("", DataType.FLOAT)
assert zp.available() == 19
with pytest.raises(LookupError):
zp.allocate("", DataType.FLOAT) # can't allocate because no more sequential bytes, only fragmented
for _ in range(14):
zp.allocate("", DataType.BYTE)
zp.allocate("", DataType.WORD)
zp.allocate("", DataType.WORD)
with pytest.raises(LookupError):
zp.allocate("", DataType.WORD)
assert zp.available() == 1
zp.allocate("last", DataType.BYTE)
with pytest.raises(LookupError):
zp.allocate("impossible", DataType.BYTE)