def tag_loop_label(self, tagger): label = tagger.make_label(self.label) tagger.push_label('times', label) body = tagger.tag_loop_label(self.body) tagger.pop_label('times') i = Var('loop_i') start_condition = (if_(eq(i,0), exit_block(label)), set(i, i-1)) body = start_condition+tuple(body)+(continue_block(label),) return begin(set(i, self.times), block(label, *body), None)
def test_eq_le_ne(self): from dao.builtins.arith import eq, le, ne eq_(eval(le(1, 1)&ne(1, 2)), True) eq_(eval(eq(1, 1)), True) eq_(eval(le(1, 1)), True)
def test_unwind_protect_loop(self): eq_(eval(let([(i,3)], block(a, set(i, sub(i, 1)), if_(eq(i, 0), exit_block(a, 1)), unwind_protect(continue_block(a), prin(2))), i)), 0)
def testLoopUntil(self): eq_(eval(tag_loop_label(let([(i,3)], LoopUntilForm((set(i, sub(i, 1)), prin(i)), eq(i,0))))), None)
def testletroddeven(self): eq_(eval(letr([(odd, lambda_([n], if_(eq(n,0), 0, even(sub(n,1))))), (even, lambda_([n], if_(eq(n,0), 1, odd(sub(n, 1)))))], odd(3))), 1)
def testloop(self): eq_(eval(let([(i,3)], block(a, set(i, sub(i, 1)), if_(eq(i, 0), exit_block(a, 1)), continue_block(a)), i)), 0)
def testletr(self): eq_(eval(letr([(f, lambda_([n], if_(eq(n, 1), 1, f(sub(n, 1)))))], f(2))), 1)
def testletrfac(self): eq_(eval(letr([(fac, lambda_([n], if_(eq(n,1), 1, mul(n, fac(sub(n, 1))))))], fac(3))), 6)
def test_iff1(self): assert_raises(DinpySyntaxError, preparse, iff(v.i == 1)[2]) eq_(preparse(iff(v.i == 1).do[2]), special.iff([(arith.eq(i, 1), 2)]))
def test_loop_until(self): eq_(preparse(loop[prin(1)].until(v.i == 1)), special.LoopUntilForm((prin(1), ), arith.eq(i, 1)))
def test_eq_le_ne(self): from dao.builtins.arith import eq, le, ne eq_(eval(le(1, 1) & ne(1, 2)), True) eq_(eval(eq(1, 1)), True) eq_(eval(le(1, 1)), True)
def test_iff1(self): assert_raises(DinpySyntaxError, preparse, iff(v.i==1)[2]) eq_(preparse(iff(v.i==1).do[2]), special.iff([(arith.eq(i,1), 2)]))
def test_loop_until(self): eq_(preparse(loop[prin(1)].until(v.i==1)), special.LoopUntilForm((prin(1),), arith.eq(i, 1)))