.section data $0x0 {
    number: .u8 $10

    main:
        ; Test addition
        mov %status, $1
        add %r0, $1
        cmpeq %r0, $1
        jz end

        ; Test addition from a register
        mov %status, $2
        add %r0, %r0
        cmpeq %r0, $2
        jz end
        
        ; Test addition from a u8 location
        mov %status, $3
        add %r0, (number)u8
        cmpeq %r0, $12
        jz end

        ; Test addition overflow
        mov %status, $4
        add %r0, $0xFFFFFFFFFFFFFFFF
        cmpeq %r0, $11
        jz end

        ; Test subtraction
        mov %status, $5
        mov %r0, (number)u8
        sub %r0, $1
        cmpeq %r0, $9
        jz end

        ; Test subtraction from a register
        mov %status, $6
        mov %r1, $2
        sub %r0, %r1
        cmpeq %r0, $7
        jz end
        
        ; Test subtraction overflow
        mov %status, $7
        sub %r0, $8
        cmpeq %r0, $0xFFFFFFFFFFFFFFFF
        jz end

        ; Test multiplication
        mov %status, $8
        mov %r0, (number)u8
        mul %r0, $1
        cmpeq %r0, $10
        jz end

        ; Test multiplication from a register
        mov %status, $9
        mov %r1, $2
        mul %r0, %r1
        cmpeq %r0, $20
        jz end
        
        ; Test negative multiplication
        mov %status, $10
        mul %r0, $0xFFFFFFFFFFFFFFFF        ; -1
        cmpeq %r0, $0xFFFFFFFFFFFFFFEC      ; -20
        jz end

        ; Test division
        mov %status, $11
        mov %r0, (number)u8
        div %r0, $2
        cmpeq %r0, $5
        jz end

        ; Test integer division
        mov %status, $12
        mov %r1, $2
        div %r0, %r1
        cmpeq %r0, $2
        jz end
        
        ; Test negative division with idiv
        mov %status, $13
        idiv %r0, $0xFFFFFFFFFFFFFFFF        ; -1
        cmpeq %r0, $0xFFFFFFFFFFFFFFFE      ; -2
        jz end

        ; Test modulo
        mov %status, $14
        mov %r0, (number)u8
        mod %r0, $4
        cmpeq %r0, $2
        jz end

        ; Test mod from a register
        mov %status, $15
        mod %r0, %r0
        cmpeq %r0, $0
        jz end
        
        ; TODO : Test mod and div by zero

        mov %status, $0
    end:
        halt

    .export main
}

.meta {
    entry: main
}