diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2017-09-13 13:33:00 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-09-13 13:33:00 -0500 |
| commit | f7be36b05feff7f54b3e118a9af265d0b9de94e2 (patch) | |
| tree | 42017722635f4cd93275488e32e395d2e3f7fc65 /numpy/linalg/tests/test_linalg.py | |
| parent | 03f3789efe4da2c56d2841ed027ef6735ca2f11b (diff) | |
| parent | ebe2cfb68586208bb096a575a603d00da5ee3887 (diff) | |
| download | numpy-f7be36b05feff7f54b3e118a9af265d0b9de94e2.tar.gz | |
Merge pull request #8827 from eric-wieser/fix-pinv
BUG: Fix pinv for stacked matrices
Diffstat (limited to 'numpy/linalg/tests/test_linalg.py')
| -rw-r--r-- | numpy/linalg/tests/test_linalg.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/numpy/linalg/tests/test_linalg.py b/numpy/linalg/tests/test_linalg.py index ab81fc485..fa20cc5ea 100644 --- a/numpy/linalg/tests/test_linalg.py +++ b/numpy/linalg/tests/test_linalg.py @@ -712,12 +712,16 @@ class TestCondInf(object): assert_almost_equal(linalg.cond(A, inf), 3.) -class TestPinv(LinalgSquareTestCase, LinalgNonsquareTestCase): +class TestPinv(LinalgSquareTestCase, + LinalgNonsquareTestCase, + LinalgGeneralizedSquareTestCase, + LinalgGeneralizedNonsquareTestCase): def do(self, a, b, tags): a_ginv = linalg.pinv(a) # `a @ a_ginv == I` does not hold if a is singular - assert_almost_equal(dot(a, a_ginv).dot(a), a, single_decimal=5, double_decimal=11) + dot = dot_generalized + assert_almost_equal(dot(dot(a, a_ginv), a), a, single_decimal=5, double_decimal=11) assert_(imply(isinstance(a, matrix), isinstance(a_ginv, matrix))) |