def testWhileCondConstant(self):
    out = lax.while_loop((lambda _: False), (lambda _: ()), ())
    self.assertEqual(out, ())
def testWhileCondConstant(self):
    out = lax.while_loop((lambda _: False), (lambda _: ()), ())
    self.assertEqual(out, ())