Like `np.testing.assert_array_equal`, but for duckarrays
(x, y, err_msg="", verbose=True)
| 310 | |
| 311 | @ensure_warnings |
| 312 | def assert_duckarray_equal(x, y, err_msg="", verbose=True): |
| 313 | """Like `np.testing.assert_array_equal`, but for duckarrays""" |
| 314 | __tracebackhide__ = True |
| 315 | |
| 316 | if not utils.is_duck_array(x) and not utils.is_scalar(x): |
| 317 | x = np.asarray(x) |
| 318 | |
| 319 | if not utils.is_duck_array(y) and not utils.is_scalar(y): |
| 320 | y = np.asarray(y) |
| 321 | |
| 322 | if (utils.is_duck_array(x) and utils.is_scalar(y)) or ( |
| 323 | utils.is_scalar(x) and utils.is_duck_array(y) |
| 324 | ): |
| 325 | equiv = duck_array_ops.array_all(x == y) |
| 326 | else: |
| 327 | equiv = duck_array_ops.array_equiv(x, y) |
| 328 | assert equiv, _format_message(x, y, err_msg=err_msg, verbose=verbose) |
| 329 | |
| 330 | |
| 331 | def assert_chunks_equal(a, b): |
searching dependent graphs…