mirror of
https://github.com/irmen/prog8.git
synced 2024-11-26 11:49:22 +00:00
91 lines
3.8 KiB
Kotlin
91 lines
3.8 KiB
Kotlin
package prog8tests.ast
|
|
|
|
import io.kotest.assertions.fail
|
|
import io.kotest.core.spec.style.FunSpec
|
|
import io.kotest.matchers.ints.shouldBeGreaterThan
|
|
import io.kotest.matchers.shouldBe
|
|
import prog8.code.ast.*
|
|
import prog8.code.core.DataType
|
|
import prog8.code.core.Position
|
|
import prog8.code.target.C64Target
|
|
import prog8.compiler.astprocessing.IntermediateAstMaker
|
|
import prog8tests.helpers.ErrorReporterForTests
|
|
import prog8tests.helpers.compileText
|
|
|
|
class TestIntermediateAst: FunSpec({
|
|
|
|
test("creation") {
|
|
val text="""
|
|
%import textio
|
|
%import graphics
|
|
main {
|
|
sub start() {
|
|
ubyte cc
|
|
ubyte[] array = [1,2,3]
|
|
cc = 11 in array
|
|
cc = sqrt(lsb(cc))
|
|
}
|
|
}
|
|
"""
|
|
val target = C64Target()
|
|
val errors = ErrorReporterForTests()
|
|
val result = compileText(target, false, text, writeAssembly = false)!!
|
|
val ast = IntermediateAstMaker(result.compilerAst, errors).transform()
|
|
ast.name shouldBe result.compilerAst.name
|
|
ast.allBlocks().any() shouldBe true
|
|
val entry = ast.entrypoint() ?: fail("no main.start() found")
|
|
entry.children.size shouldBe 5
|
|
entry.name shouldBe "start"
|
|
entry.scopedName shouldBe "main.start"
|
|
val blocks = ast.allBlocks().toList()
|
|
blocks.size shouldBeGreaterThan 1
|
|
blocks[0].name shouldBe "main"
|
|
blocks[0].scopedName shouldBe "main"
|
|
|
|
val ccInit = entry.children[2] as PtAssignment
|
|
ccInit.target.identifier?.name shouldBe "main.start.cc"
|
|
(ccInit.value as PtNumber).number shouldBe 0.0
|
|
|
|
val ccdecl = entry.children[0] as PtVariable
|
|
ccdecl.name shouldBe "cc"
|
|
ccdecl.scopedName shouldBe "main.start.cc"
|
|
ccdecl.type shouldBe DataType.UBYTE
|
|
val arraydecl = entry.children[1] as IPtVariable
|
|
arraydecl.name shouldBe "array"
|
|
arraydecl.type shouldBe DataType.ARRAY_UB
|
|
|
|
val containment = (entry.children[3] as PtAssignment).value as PtContainmentCheck
|
|
(containment.element as PtNumber).number shouldBe 11.0
|
|
val fcall = (entry.children[4] as PtAssignment).value as PtFunctionCall
|
|
fcall.void shouldBe false
|
|
fcall.type shouldBe DataType.UBYTE
|
|
}
|
|
|
|
test("isSame on binaryExpressions") {
|
|
val expr1 = PtBinaryExpression("/", DataType.UBYTE, Position.DUMMY)
|
|
expr1.add(PtNumber(DataType.UBYTE, 1.0, Position.DUMMY))
|
|
expr1.add(PtNumber(DataType.UBYTE, 2.0, Position.DUMMY))
|
|
val expr2 = PtBinaryExpression("/", DataType.UBYTE, Position.DUMMY)
|
|
expr2.add(PtNumber(DataType.UBYTE, 1.0, Position.DUMMY))
|
|
expr2.add(PtNumber(DataType.UBYTE, 2.0, Position.DUMMY))
|
|
(expr1 isSameAs expr2) shouldBe true
|
|
val expr3 = PtBinaryExpression("/", DataType.UBYTE, Position.DUMMY)
|
|
expr3.add(PtNumber(DataType.UBYTE, 2.0, Position.DUMMY))
|
|
expr3.add(PtNumber(DataType.UBYTE, 1.0, Position.DUMMY))
|
|
(expr1 isSameAs expr3) shouldBe false
|
|
}
|
|
|
|
test("isSame on binaryExpressions with associative operators") {
|
|
val expr1 = PtBinaryExpression("+", DataType.UBYTE, Position.DUMMY)
|
|
expr1.add(PtNumber(DataType.UBYTE, 1.0, Position.DUMMY))
|
|
expr1.add(PtNumber(DataType.UBYTE, 2.0, Position.DUMMY))
|
|
val expr2 = PtBinaryExpression("+", DataType.UBYTE, Position.DUMMY)
|
|
expr2.add(PtNumber(DataType.UBYTE, 1.0, Position.DUMMY))
|
|
expr2.add(PtNumber(DataType.UBYTE, 2.0, Position.DUMMY))
|
|
(expr1 isSameAs expr2) shouldBe true
|
|
val expr3 = PtBinaryExpression("+", DataType.UBYTE, Position.DUMMY)
|
|
expr3.add(PtNumber(DataType.UBYTE, 2.0, Position.DUMMY))
|
|
expr3.add(PtNumber(DataType.UBYTE, 1.0, Position.DUMMY))
|
|
(expr1 isSameAs expr3) shouldBe true
|
|
}
|
|
}) |