| 459 | |
| 460 | |
| 461 | def test_divmod(): |
| 462 | arr1 = np.random.randint(1, 100, size=(20, 20)) |
| 463 | arr2 = np.random.randint(1, 100, size=(20, 20)) |
| 464 | |
| 465 | darr1 = da.from_array(arr1, 3) |
| 466 | darr2 = da.from_array(arr2, 3) |
| 467 | |
| 468 | result = np.divmod(darr1, 2.0) |
| 469 | expected = np.divmod(arr1, 2.0) |
| 470 | assert_eq(result[0], expected[0]) |
| 471 | assert_eq(result[1], expected[1]) |
| 472 | |
| 473 | result = np.divmod(darr1, darr2) |
| 474 | expected = np.divmod(arr1, arr2) |
| 475 | assert_eq(result[0], expected[0]) |
| 476 | assert_eq(result[1], expected[1]) |
| 477 | |
| 478 | result = divmod(darr1, 2.0) |
| 479 | expected = divmod(arr1, 2.0) |
| 480 | assert_eq(result[0], expected[0]) |
| 481 | assert_eq(result[1], expected[1]) |
| 482 | |
| 483 | result = divmod(darr1, darr2) |
| 484 | expected = divmod(arr1, arr2) |
| 485 | assert_eq(result[0], expected[0]) |
| 486 | assert_eq(result[1], expected[1]) |
| 487 | |
| 488 | |
| 489 | @pytest.mark.parametrize("dt", ["float64", "float32", "int32", "int64"]) |