diff options
| author | Pauli Virtanen <pav@iki.fi> | 2017-01-29 22:06:05 +0100 |
|---|---|---|
| committer | Pauli Virtanen <pav@iki.fi> | 2017-01-29 22:06:49 +0100 |
| commit | 24e551b56c5af03e0cbfdb90a30ff299711094f0 (patch) | |
| tree | 95bf9ba5710e2f384f6e7f82708b820cf89da1cc /numpy | |
| parent | 780ef22557f25e1531221e06657afc18af084117 (diff) | |
| download | numpy-24e551b56c5af03e0cbfdb90a30ff299711094f0.tar.gz | |
TST: core: test all dtypes in test_dot_out_mem_overlap
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index 1ef3bfad7..3bbe99b8f 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -2329,9 +2329,11 @@ class TestMethods(TestCase): def test_dot_out_mem_overlap(self): np.random.seed(1) - # Test BLAS and non-BLAS code paths - for dtype in [np.float32, np.int64]: - + # Test BLAS and non-BLAS code paths, including all dtypes + # that dot() supports + dtypes = [np.dtype(code) for code in np.typecodes['All'] + if code not in 'USVM'] + for dtype in dtypes: a = np.random.rand(3, 3).astype(dtype) b = np.random.rand(3, 3).astype(dtype) y = np.dot(a, b) |