def test_num_regs(): j = JitCode("test") j.setup(num_regs_i=12, num_regs_r=34, num_regs_f=56) assert j.num_regs_i() == 12 assert j.num_regs_r() == 34 assert j.num_regs_f() == 56 j.setup(num_regs_i=0, num_regs_r=0, num_regs_f=0) assert j.num_regs_i() == 0 assert j.num_regs_r() == 0 assert j.num_regs_f() == 0 j.setup(num_regs_i=255, num_regs_r=255, num_regs_f=255) assert j.num_regs_i() == 255 assert j.num_regs_r() == 255 assert j.num_regs_f() == 255