def test_common_subexprs_elimination(tu, tv, tw, ti0, ti1, t0, t1, exprs, expected): counter = generator() make = lambda: Scalar(name='r%d' % counter()).indexify() processed = common_subexprs_elimination(EVAL(exprs, tu, tv, tw, ti0, ti1, t0, t1), make) assert len(processed) == len(expected) assert all(str(i.rhs) == j for i, j in zip(processed, expected))
def test_common_subexprs_elimination(tu, tv, tw, ti0, ti1, t0, t1, exprs, expected): make = lambda i: Scalar(name='r%d' % i).indexify() processed = common_subexprs_elimination( EVAL(exprs, tu, tv, tw, ti0, ti1, t0, t1), make) assert len(processed) == len(expected) assert all(str(i.rhs) == j for i, j in zip(processed, expected))
def test_common_subexprs_elimination(tu, tv, tw, ti0, ti1, t0, t1, exprs, expected): processed = common_subexprs_elimination( EVAL(exprs, tu, tv, tw, ti0, ti1, t0, t1), lambda i: Symbol('r%d' % i)) assert len(processed) == len(expected) assert all(str(i.rhs) == j for i, j in zip(processed, expected))