diff options
| author | Toshiki Kataoka <kataoka@preferred.jp> | 2018-04-17 21:17:36 +0900 |
|---|---|---|
| committer | Toshiki Kataoka <kataoka@preferred.jp> | 2018-04-17 21:17:36 +0900 |
| commit | b8d6d4d35feef83844ac2b067c3d293990bff4f5 (patch) | |
| tree | f029a2223784c6e722d3635482a151fefd842d0e | |
| parent | 3e595509e6e237259ce0e934ada720dac6e3ac13 (diff) | |
| download | numpy-b8d6d4d35feef83844ac2b067c3d293990bff4f5.tar.gz | |
add tests
| -rw-r--r-- | numpy/core/tests/test_einsum.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/numpy/core/tests/test_einsum.py b/numpy/core/tests/test_einsum.py index 792b9e0a2..104dd1986 100644 --- a/numpy/core/tests/test_einsum.py +++ b/numpy/core/tests/test_einsum.py @@ -502,6 +502,16 @@ class TestEinSum(object): optimize=optimize), np.full((1, 5), 5)) + # Cases which were failing (gh-10899) + x = np.eye(2, dtype=dtype) + y = np.ones(2, dtype=dtype) + assert_array_equal(np.einsum("ji,i->", x, y, optimize=optimize), + [2.]) # contig_contig_outstride0_two + assert_array_equal(np.einsum("i,ij->", y, x, optimize=optimize), + [2.]) # stride0_contig_outstride0_two + assert_array_equal(np.einsum("ij,i->", x, y, optimize=optimize), + [2.]) # contig_stride0_outstride0_two + def test_einsum_sums_int8(self): self.check_einsum_sums('i1') |