def testIssue744(self):
    Point = collections.namedtuple('Point', ['x', 'y'])
    p0 = Point(x=jnp.array(1), y=jnp.array(2))

    def plus_one(p, iter_idx):
        return (Point((p.x + 1), (p.y + 1)), iter_idx)
    self.assertRaisesRegex(ValueError, 'scan got value with no leading axis to scan over.*', (lambda : lax.scan(plus_one, p0, list(range(5)))))
def testIssue744(self):
    Point = collections.namedtuple('Point', ['x', 'y'])
    p0 = Point(x=jnp.array(1), y=jnp.array(2))

    def plus_one(p, iter_idx):
        return (Point((p.x + 1), (p.y + 1)), iter_idx)
    self.assertRaisesRegex(ValueError, 'scan got value with no leading axis to scan over.*', (lambda : lax.scan(plus_one, p0, list(range(5)))))