from coparun_module import coparun
from copapy import Write, CPVariable
import copapy
from copapy import binwrite


def test_compile():

    c1 = CPVariable(4)
    c2 = CPVariable(2) * 4

    i1 = c2 * 2
    r1 = i1 + 7 + (c1 + 7 * 9)
    r2 = i1 + 9
    out = [Write(r1), Write(r2), Write(c2)]

    il, _ = copapy.compile_to_instruction_list(out, copapy.generic_sdb)

    # run program command
    il.write_com(binwrite.Command.RUN_PROG)

    il.write_com(binwrite.Command.READ_DATA)
    il.write_int(0)
    il.write_int(36)

    # run program command
    il.write_com(binwrite.Command.END_COM)

    #print('* Data to runner:')
    #il.print()

    print('+ run coparun')
    result = coparun(il.get_data())

    assert result == 1


if __name__ == "__main__":
    test_compile()