コード例 #1
0
def test_no_2d_strided_vector():
    a_orig = numpy.ones((10, 10), dtype=float)
    a = a_orig[:3, :3]

    try:
        te.dbl_numpy_strided_vec_inplace(a)
        assert False
    except ValueError:
        pass
コード例 #2
0
def test_no_2d_strided_vector():
    a_orig = numpy.ones((10, 10), dtype=float)
    a = a_orig[:3, :3]

    try:
        te.dbl_numpy_strided_vec_inplace(a)
        assert False
    except ValueError:
        pass
コード例 #3
0
def test_vec_slice_noncontig_inplace():
    a_orig = numpy.ones((10, ), dtype=float)
    a = a_orig[::2]

    te.dbl_numpy_vec_inplace(a)
    assert (a_orig == 2).all()

    a_orig = numpy.ones((10, ), dtype=float)
    a = a_orig[::2]

    te.dbl_numpy_strided_vec_inplace(a)
    assert (a_orig[::2] == 2).all()
    assert (a_orig[1::2] == 1).all()
コード例 #4
0
def test_vec_slice_noncontig_inplace():
    a_orig = numpy.ones((10,), dtype=float)
    a = a_orig[::2]

    te.dbl_numpy_vec_inplace(a)
    assert (a_orig == 2).all()

    a_orig = numpy.ones((10,), dtype=float)
    a = a_orig[::2]

    te.dbl_numpy_strided_vec_inplace(a)
    assert (a_orig[::2] == 2).all()
    assert (a_orig[1::2] == 1).all()
コード例 #5
0
def test_negative_stride():
    a_orig = numpy.ones((10, ), dtype=float)
    a = a_orig[::-2]

    te.dbl_numpy_vec_inplace(a)
    assert (a_orig[1:] == 2).all()
    assert (a_orig[0] == 1).all()

    a_orig = numpy.ones((10, ), dtype=float)
    a = a_orig[::-2]

    te.dbl_numpy_strided_vec_inplace(a)
    assert (a_orig[::-2] == 2).all()
    assert (a_orig[-2::-2] == 1).all()
コード例 #6
0
def test_negative_stride():
    a_orig = numpy.ones((10,), dtype=float)
    a = a_orig[::-2]

    te.dbl_numpy_vec_inplace(a)
    assert (a_orig[1:] == 2).all()
    assert (a_orig[0] == 1).all()

    a_orig = numpy.ones((10,), dtype=float)
    a = a_orig[::-2]

    te.dbl_numpy_strided_vec_inplace(a)
    assert (a_orig[::-2] == 2).all()
    assert (a_orig[-2::-2] == 1).all()