diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2017-04-12 18:31:18 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-04-12 18:31:18 -0600 |
| commit | 0f3846aaabc7c1278e9aeab7aafa0b8d4a1b264e (patch) | |
| tree | 0d5d9dbc72cdc1f9edb9a420f40aef448da714cc /numpy/linalg/tests | |
| parent | 7843711cbe830c330eebf23622021132d20e73e0 (diff) | |
| parent | ad8d6eb4942c68a8123a94d5dd73bdb246775690 (diff) | |
| download | numpy-0f3846aaabc7c1278e9aeab7aafa0b8d4a1b264e.tar.gz | |
Merge pull request #8682 from eric-wieser/stack-matrix_rank
ENH: allow argument to matrix_rank to be stacked
Diffstat (limited to 'numpy/linalg/tests')
| -rw-r--r-- | numpy/linalg/tests/test_linalg.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/numpy/linalg/tests/test_linalg.py b/numpy/linalg/tests/test_linalg.py index 795a28954..baa195241 100644 --- a/numpy/linalg/tests/test_linalg.py +++ b/numpy/linalg/tests/test_linalg.py @@ -1246,8 +1246,9 @@ class TestMatrixRank(object): yield assert_equal, matrix_rank(np.zeros((4,))), 0 # accepts array-like yield assert_equal, matrix_rank([1]), 1 - # greater than 2 dimensions raises error - yield assert_raises, TypeError, matrix_rank, np.zeros((2, 2, 2)) + # greater than 2 dimensions treated as stacked matrices + ms = np.array([I, np.eye(4), np.zeros((4,4))]) + yield assert_equal, matrix_rank(ms), np.array([3, 4, 0]) # works on scalar yield assert_equal, matrix_rank(1), 1 |