()
| 240 | # Gufunc test cases |
| 241 | # |
| 242 | def _make_generalized_cases(): |
| 243 | new_cases = [] |
| 244 | |
| 245 | for case in CASES: |
| 246 | if not isinstance(case.a, np.ndarray): |
| 247 | continue |
| 248 | |
| 249 | a = np.array([case.a, 2 * case.a, 3 * case.a]) |
| 250 | if case.b is None: |
| 251 | b = None |
| 252 | else: |
| 253 | b = np.array([case.b, 7 * case.b, 6 * case.b]) |
| 254 | new_case = LinalgCase(case.name + "_tile3", a, b, |
| 255 | tags=case.tags | {'generalized'}) |
| 256 | new_cases.append(new_case) |
| 257 | |
| 258 | a = np.array([case.a] * 2 * 3).reshape((3, 2) + case.a.shape) |
| 259 | if case.b is None: |
| 260 | b = None |
| 261 | else: |
| 262 | b = np.array([case.b] * 2 * 3).reshape((3, 2) + case.b.shape) |
| 263 | new_case = LinalgCase(case.name + "_tile213", a, b, |
| 264 | tags=case.tags | {'generalized'}) |
| 265 | new_cases.append(new_case) |
| 266 | |
| 267 | return new_cases |
| 268 | |
| 269 | |
| 270 | CASES += _make_generalized_cases() |
no test coverage detected