TestJit.test_pretty_printer-while_test.expect

def script(a1, i1):
  t4 = aten::lt(i1, 3)
  t5 = prim::TensorToBool(t4)
  a = a1
  i = i1
  t6 = t5
  while t6:
    i2 = aten::add(i, 1, 1)
    t13 = aten::lt(i2, 3)
    t6 = prim::TensorToBool(t13)
    a = aten::mul(a, a)
    i = i2
  return a