summaryrefslogtreecommitdiff
path: root/numpy/core/tests/test_einsum.py
diff options
context:
space:
mode:
authorEric Wieser <wieser.eric@gmail.com>2018-03-26 00:55:10 -0700
committerGitHub <noreply@github.com>2018-03-26 00:55:10 -0700
commit342f0e34c4921a20f193c339e4ca15e0d70a10ae (patch)
tree655576a814ecee9b7a6c2ec508d41e85b68707a5 /numpy/core/tests/test_einsum.py
parenteca4175abcdff3fc506992c74fbab67aeb48153e (diff)
parenta1146a9a473b7dc229fe457a383afd5394cd5cf2 (diff)
downloadpython-numpy-342f0e34c4921a20f193c339e4ca15e0d70a10ae.tar.gz
python-numpy-342f0e34c4921a20f193c339e4ca15e0d70a10ae.tar.bz2
python-numpy-342f0e34c4921a20f193c339e4ca15e0d70a10ae.zip
Merge pull request #10798 from jaimefrio/einsum_mapping
BUG: error checking before mapping of einsum axes.
Diffstat (limited to 'numpy/core/tests/test_einsum.py')
-rw-r--r--numpy/core/tests/test_einsum.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/numpy/core/tests/test_einsum.py b/numpy/core/tests/test_einsum.py
index 9bd85fdb9..5863c11ad 100644
--- a/numpy/core/tests/test_einsum.py
+++ b/numpy/core/tests/test_einsum.py
@@ -791,6 +791,12 @@ class TestEinSum(object):
self.optimize_compare('dba,ead,cad->bce')
self.optimize_compare('aef,fbc,dca->bde')
+ def test_combined_views_mapping(self):
+ # gh-10792
+ a = np.arange(9).reshape(1, 1, 3, 1, 3)
+ b = np.einsum('bbcdc->d', a)
+ assert_equal(b, [12])
+
class TestEinSumPath(object):
def build_operands(self, string, size_dict=global_size_dict):