(self)
| 129 | |
| 130 | class TestDiag: |
| 131 | def test_vector(self): |
| 132 | vals = (100 * arange(5)).astype('l') |
| 133 | b = zeros((5, 5)) |
| 134 | for k in range(5): |
| 135 | b[k, k] = vals[k] |
| 136 | assert_equal(diag(vals), b) |
| 137 | b = zeros((7, 7)) |
| 138 | c = b.copy() |
| 139 | for k in range(5): |
| 140 | b[k, k + 2] = vals[k] |
| 141 | c[k + 2, k] = vals[k] |
| 142 | assert_equal(diag(vals, k=2), b) |
| 143 | assert_equal(diag(vals, k=-2), c) |
| 144 | |
| 145 | def test_matrix(self, vals=None): |
| 146 | if vals is None: |
nothing calls this directly
no test coverage detected