0370917c6e71a01dbe1260c92e10147eac653e5d,scipy/weave/tests/test_blitz_tools.py,,test_blitz_bug,#,163

Before Change

        blitz(expr_buggy.format(lim))
        blitz(expr_not_buggy.format(lim, "sh"))
        if old_env is None:
            os.environ.pop("PYTHONCOMPILED")
        else:
            os.environ["PYTHONCOMPILED"] = old_env

        arr_np[lim:] = arr[lim:]

After Change

@dec.slow
def test_blitz_bug():
    // Assignment to arr[i:] used to fail inside blitz expressions.
    with TempdirBlitz():
        N = 4
        expr_buggy = "arr_blitz_buggy[{0}:] = arr[{0}:]"
        expr_not_buggy = "arr_blitz_not_buggy[{0}:{1}] = arr[{0}:]"
        random.seed(7)
        arr = random.randn(N)
        sh = arr.shape[0]
        for lim in [0, 1, 2]:
            arr_blitz_buggy = zeros(N)
            arr_blitz_not_buggy = zeros(N)
            arr_np = zeros(N)
            blitz(expr_buggy.format(lim))
            blitz(expr_not_buggy.format(lim, "sh"))
            arr_np[lim:] = arr[lim:]
            assert_allclose(arr_blitz_buggy, arr_np)
            assert_allclose(arr_blitz_not_buggy, arr_np)


if __name__ == "__main__":
    run_module_suite()

Instances