summaryrefslogtreecommitdiff
path: root/numpy
diff options
context:
space:
mode:
authorPauli Virtanen <pav@iki.fi>2017-01-29 16:09:53 +0100
committerPauli Virtanen <pav@iki.fi>2017-01-29 16:09:53 +0100
commitd2848f055b85ec95a3097d6763d9ff89151d81e1 (patch)
treeb407f969c4a39008bde0d9b1995597350a6f8645 /numpy
parentcacc2ed5fd6bc06f16061f6f230a592ea8c68d3e (diff)
downloadnumpy-d2848f055b85ec95a3097d6763d9ff89151d81e1.tar.gz
TST: core: make test_dot_out_mem_overlap minimal
Diffstat (limited to 'numpy')
-rw-r--r--numpy/core/tests/test_multiarray.py16
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])