mirror of
https://github.com/irmen/prog8.git
synced 2024-11-27 03:50:27 +00:00
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
import pytest
|
|
from il65.plyparse import IncrDecr, AugAssignment, VarDef, SymbolName
|
|
from il65.optimize import optimize
|
|
from .test_parser import parse_source
|
|
|
|
|
|
def test_incrdecr_joins_nonfloat():
|
|
src = """~ test {
|
|
X ++
|
|
X ++
|
|
X += 10
|
|
Y--
|
|
Y--
|
|
Y-=20
|
|
}"""
|
|
result = parse_source(src)
|
|
testscope = result.scope.nodes[0].nodes[0]
|
|
assert len(testscope.nodes) == 6
|
|
assert isinstance(testscope.nodes[0], IncrDecr)
|
|
assert testscope.nodes[0].howmuch == 1
|
|
assert isinstance(testscope.nodes[1], IncrDecr)
|
|
assert testscope.nodes[1].howmuch == 1
|
|
assert isinstance(testscope.nodes[2], AugAssignment)
|
|
assert testscope.nodes[2].right.value == 10
|
|
assert isinstance(testscope.nodes[3], IncrDecr)
|
|
assert testscope.nodes[3].howmuch == 1
|
|
assert isinstance(testscope.nodes[4], IncrDecr)
|
|
assert testscope.nodes[4].howmuch == 1
|
|
assert isinstance(testscope.nodes[5], AugAssignment)
|
|
assert testscope.nodes[5].right.value == 20
|
|
# now optimize the incrdecrs (joins them)
|
|
optimize(result)
|
|
testscope = result.scope.nodes[0].nodes[0]
|
|
assert len(testscope.nodes) == 2 # @todo broken optimization right now
|
|
assert isinstance(testscope.nodes[0], IncrDecr)
|
|
assert testscope.nodes[0].operator == "++"
|
|
assert testscope.nodes[0].howmuch == 12
|
|
assert isinstance(testscope.nodes[1], IncrDecr)
|
|
assert testscope.nodes[1].operator == "--"
|
|
assert testscope.nodes[1].howmuch == 22
|
|
|
|
|
|
def test_incrdecr_joins_float():
|
|
src = """~ test {
|
|
var .float flt = 0
|
|
flt ++
|
|
flt ++
|
|
flt += 10
|
|
flt --
|
|
flt --
|
|
flt --
|
|
flt -= 5
|
|
}"""
|
|
result = parse_source(src)
|
|
testscope = result.scope.nodes[0].nodes[0]
|
|
assert len(testscope.nodes) == 8
|
|
# now optimize the incrdecrs (joins them)
|
|
optimize(result)
|
|
testscope = result.scope.nodes[0].nodes[0]
|
|
assert len(testscope.nodes) == 2
|
|
assert isinstance(testscope.nodes[0], VarDef)
|
|
assert isinstance(testscope.nodes[1], IncrDecr)
|
|
assert testscope.nodes[1].operator == "++"
|
|
assert testscope.nodes[1].howmuch == 4
|
|
assert isinstance(testscope.nodes[1].target, SymbolName)
|
|
assert testscope.nodes[1].target.name == "flt"
|
|
|
|
|
|
def test_large_incrdecr_to_augassign():
|
|
src = """~ test {
|
|
X ++
|
|
X ++
|
|
X += 255
|
|
Y --
|
|
Y --
|
|
Y -= 255
|
|
}"""
|
|
result = parse_source(src)
|
|
testscope = result.scope.nodes[0].nodes[0]
|
|
assert len(testscope.nodes) == 6
|
|
# now optimize; joins the incrdecrs then converts to augassign because values are too large.
|
|
optimize(result)
|
|
testscope = result.scope.nodes[0].nodes[0]
|
|
assert len(testscope.nodes) == 2
|
|
assert isinstance(testscope.nodes[0], AugAssignment)
|
|
assert testscope.nodes[0].left.name == "X"
|
|
assert testscope.nodes[0].operator == "+="
|
|
assert testscope.nodes[0].right.value == 257
|
|
assert isinstance(testscope.nodes[1], AugAssignment)
|
|
assert testscope.nodes[1].left.name == "Y"
|
|
assert testscope.nodes[1].operator == "-="
|
|
assert testscope.nodes[1].right.value == 257
|