Test of take, transpose, inner, outer products.
(self)
| 212 | |
| 213 | @np.errstate(all='ignore') |
| 214 | def test_4(self): |
| 215 | """ |
| 216 | Test of take, transpose, inner, outer products. |
| 217 | |
| 218 | """ |
| 219 | x = self.arange(24) |
| 220 | y = np.arange(24) |
| 221 | x[5:6] = self.masked |
| 222 | x = x.reshape(2, 3, 4) |
| 223 | y = y.reshape(2, 3, 4) |
| 224 | assert self.allequal(np.transpose(y, (2, 0, 1)), self.transpose(x, (2, 0, 1))) |
| 225 | assert self.allequal(np.take(y, (2, 0, 1), 1), self.take(x, (2, 0, 1), 1)) |
| 226 | assert self.allequal(np.inner(self.filled(x, 0), self.filled(y, 0)), |
| 227 | self.inner(x, y)) |
| 228 | assert self.allequal(np.outer(self.filled(x, 0), self.filled(y, 0)), |
| 229 | self.outer(x, y)) |
| 230 | y = self.array(['abc', 1, 'def', 2, 3], object) |
| 231 | y[2] = self.masked |
| 232 | t = self.take(y, [0, 3, 4]) |
| 233 | assert t[0] == 'abc' |
| 234 | assert t[1] == 2 |
| 235 | assert t[2] == 3 |
| 236 | |
| 237 | @np.errstate(all='ignore') |
| 238 | def test_5(self): |