Function run_test

imperative/python/test/unit/core/test_indexing_op.py:639–645

(fn)

    mge_x = megengine.tensor(np_x)

    def run_test(fn):
        out_ref = fn(np_x)
        if symbolic is not None:
            fn = jit.trace(symbolic=symbolic)(fn)
        for i in range(3):
            out = fn(mge_x)
            np.testing.assert_equal(out.numpy(), out_ref)

    run_test(lambda x: x[0:1, :, :])
    run_test(lambda x: x[1:100:2, :, :])