(self, vals=None)
| 125 | assert_equal(diag(vals, k=-2), c) |
| 126 | |
| 127 | def test_matrix(self, vals=None): |
| 128 | if vals is None: |
| 129 | vals = (100 * get_mat(5) + 1).astype('l') |
| 130 | b = zeros((5,)) |
| 131 | for k in range(5): |
| 132 | b[k] = vals[k, k] |
| 133 | assert_equal(diag(vals), b) |
| 134 | b = b * 0 |
| 135 | for k in range(3): |
| 136 | b[k] = vals[k, k + 2] |
| 137 | assert_equal(diag(vals, 2), b[:3]) |
| 138 | for k in range(3): |
| 139 | b[k] = vals[k + 2, k] |
| 140 | assert_equal(diag(vals, -2), b[:3]) |
| 141 | |
| 142 | def test_fortran_order(self): |
| 143 | vals = array((100 * get_mat(5) + 1), order='F', dtype='l') |
no test coverage detected