diff options
author | Pauli Virtanen <pav@iki.fi> | 2017-01-29 16:09:53 +0100 |
---|---|---|
committer | Pauli Virtanen <pav@iki.fi> | 2017-01-29 16:09:53 +0100 |
commit | d2848f055b85ec95a3097d6763d9ff89151d81e1 (patch) | |
tree | b407f969c4a39008bde0d9b1995597350a6f8645 | |
parent | cacc2ed5fd6bc06f16061f6f230a592ea8c68d3e (diff) | |
download | numpy-d2848f055b85ec95a3097d6763d9ff89151d81e1.tar.gz |
TST: core: make test_dot_out_mem_overlap minimal
-rw-r--r-- | numpy/core/tests/test_multiarray.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index 2621daf37..1ef3bfad7 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -2329,14 +2329,14 @@ class TestMethods(TestCase): def test_dot_out_mem_overlap(self): np.random.seed(1) - for dtype in [np.object_, np.float32, np.complex128, np.int64]: - a0 = np.random.rand(3, 3).astype(dtype) - b0 = np.random.rand(3, 3).astype(dtype) - for a, b in [(a0.copy(), b0.copy()), - (a0.copy().T, b0.copy())]: - y = np.dot(a, b) - x = np.dot(a, b, out=b) - assert_equal(x, y, err_msg=repr(dtype)) + # Test BLAS and non-BLAS code paths + for dtype in [np.float32, np.int64]: + + a = np.random.rand(3, 3).astype(dtype) + b = np.random.rand(3, 3).astype(dtype) + y = np.dot(a, b) + x = np.dot(a, b, out=b) + assert_equal(x, y, err_msg=repr(dtype)) # Check invalid output array assert_raises(ValueError, np.dot, a, b, out=b[::2]) |