package prog8tests import io.kotest.core.spec.style.FunSpec import io.kotest.matchers.shouldBe import io.kotest.matchers.shouldNotBe import prog8.code.core.DataType import prog8.code.core.NumericDatatypes import prog8.code.core.RegisterOrPair import prog8.code.target.C64Target import prog8.code.target.Cx16Target import prog8.compiler.BuiltinFunctions import prog8tests.helpers.compileText class TestBuiltinFunctions: FunSpec({ test("push pop") { val src=""" main { sub start () { pushw(cx16.r0) push(cx16.r1L) pop(cx16.r1L) popw(cx16.r0) } }""" compileText(Cx16Target(), false, src, writeAssembly = true) shouldNotBe null } test("pure func with fixed type") { val func = BuiltinFunctions.getValue("sgn") func.name shouldBe "sgn" func.parameters.size shouldBe 1 func.parameters[0].name shouldBe "value" func.parameters[0].possibleDatatypes shouldBe NumericDatatypes func.pure shouldBe true func.returnType shouldBe DataType.BYTE val conv = func.callConvention(listOf(DataType.UBYTE)) conv.params.size shouldBe 1 conv.params[0].dt shouldBe DataType.UBYTE conv.params[0].reg shouldBe RegisterOrPair.A conv.params[0].variable shouldBe false conv.returns.dt shouldBe DataType.BYTE conv.returns.floatFac1 shouldBe false conv.returns.reg shouldBe RegisterOrPair.A } test("not-pure func with fixed type") { val func = BuiltinFunctions.getValue("rnd") func.name shouldBe "rnd" func.parameters.size shouldBe 0 func.pure shouldBe false func.returnType shouldBe DataType.UBYTE val conv = func.callConvention(emptyList()) conv.params.size shouldBe 0 conv.returns.dt shouldBe DataType.UBYTE conv.returns.floatFac1 shouldBe false conv.returns.reg shouldBe RegisterOrPair.A } test("func without return type") { val func = BuiltinFunctions.getValue("poke") func.name shouldBe "poke" func.parameters.size shouldBe 2 func.parameters[0].name shouldBe "address" func.parameters[0].possibleDatatypes shouldBe arrayOf(DataType.UWORD) func.parameters[1].name shouldBe "value" func.parameters[1].possibleDatatypes shouldBe arrayOf(DataType.UBYTE) func.pure shouldBe false func.returnType shouldBe null val conv = func.callConvention(listOf(DataType.UWORD, DataType.UBYTE)) conv.params.size shouldBe 2 conv.params[0].dt shouldBe DataType.UWORD conv.params[0].reg shouldBe null conv.params[0].variable shouldBe true conv.params[1].dt shouldBe DataType.UBYTE conv.params[1].reg shouldBe null conv.params[1].variable shouldBe true conv.returns.dt shouldBe null conv.returns.floatFac1 shouldBe false conv.returns.reg shouldBe null } })