From 6a119c982463097aebc8d004fc731b424721a650 Mon Sep 17 00:00:00 2001 From: Eric Wieser Date: Tue, 1 Aug 2017 17:39:11 +0000 Subject: BUG: Be explicit about allowing all integer types in matrix exponentiation Fixes gh-9506, unsigned exponentiation --- numpy/matrixlib/tests/test_defmatrix.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'numpy/matrixlib/tests') diff --git a/numpy/matrixlib/tests/test_defmatrix.py b/numpy/matrixlib/tests/test_defmatrix.py index e4c3c49fb..77f262031 100644 --- a/numpy/matrixlib/tests/test_defmatrix.py +++ b/numpy/matrixlib/tests/test_defmatrix.py @@ -249,6 +249,12 @@ class TestAlgebra(object): assert_array_almost_equal(m4, np.dot(m2, m2)) assert_array_almost_equal(np.dot(mi, m), np.eye(2)) + def test_scalar_type_pow(self): + m = matrix([[1, 2], [3, 4]]) + for scalar_t in [np.int8, np.uint8]: + two = scalar_t(2) + assert_array_almost_equal(m ** 2, m ** two) + def test_notimplemented(self): '''Check that 'not implemented' operations produce a failure.''' A = matrix([[1., 2.], -- cgit v1.2.1