(a, b, check_ndim=True, check_nan=True)
| 230 | |
| 231 | |
| 232 | def assert_eq_shape(a, b, check_ndim=True, check_nan=True): |
| 233 | if check_ndim: |
| 234 | assert len(a) == len(b) |
| 235 | |
| 236 | for aa, bb in zip(a, b): |
| 237 | if math.isnan(aa) or math.isnan(bb): |
| 238 | if check_nan: |
| 239 | assert math.isnan(aa) == math.isnan(bb) |
| 240 | else: |
| 241 | assert aa == bb |
| 242 | |
| 243 | |
| 244 | def _check_chunks(x, check_ndim=True, scheduler=None): |
no outgoing calls
no test coverage detected