| 2594 | |
| 2595 | |
| 2596 | def test_arithmetic(): |
| 2597 | x = np.arange(5).astype("f4") + 2 |
| 2598 | y = np.arange(5).astype("i8") + 2 |
| 2599 | z = np.arange(5).astype("i4") + 2 |
| 2600 | a = da.from_array(x, chunks=(2,)) |
| 2601 | b = da.from_array(y, chunks=(2,)) |
| 2602 | c = da.from_array(z, chunks=(2,)) |
| 2603 | assert_eq(a + b, x + y) |
| 2604 | assert_eq(a * b, x * y) |
| 2605 | assert_eq(a - b, x - y) |
| 2606 | assert_eq(a / b, x / y) |
| 2607 | assert_eq(b & b, y & y) |
| 2608 | assert_eq(b | b, y | y) |
| 2609 | assert_eq(b ^ b, y ^ y) |
| 2610 | assert_eq(a // b, x // y) |
| 2611 | assert_eq(a**b, x**y) |
| 2612 | assert_eq(a % b, x % y) |
| 2613 | assert_eq(a > b, x > y) |
| 2614 | assert_eq(a < b, x < y) |
| 2615 | assert_eq(a >= b, x >= y) |
| 2616 | assert_eq(a <= b, x <= y) |
| 2617 | assert_eq(a == b, x == y) |
| 2618 | assert_eq(a != b, x != y) |
| 2619 | |
| 2620 | assert_eq(a + 2, x + 2) |
| 2621 | assert_eq(a * 2, x * 2) |
| 2622 | assert_eq(a - 2, x - 2) |
| 2623 | assert_eq(a / 2, x / 2) |
| 2624 | assert_eq(b & True, y & True) |
| 2625 | assert_eq(b | True, y | True) |
| 2626 | assert_eq(b ^ True, y ^ True) |
| 2627 | assert_eq(a // 2, x // 2) |
| 2628 | assert_eq(a**2, x**2) |
| 2629 | assert_eq(a % 2, x % 2) |
| 2630 | assert_eq(a > 2, x > 2) |
| 2631 | assert_eq(a < 2, x < 2) |
| 2632 | assert_eq(a >= 2, x >= 2) |
| 2633 | assert_eq(a <= 2, x <= 2) |
| 2634 | assert_eq(a == 2, x == 2) |
| 2635 | assert_eq(a != 2, x != 2) |
| 2636 | |
| 2637 | assert_eq(2 + b, 2 + y) |
| 2638 | assert_eq(2 * b, 2 * y) |
| 2639 | assert_eq(2 - b, 2 - y) |
| 2640 | assert_eq(2 / b, 2 / y) |
| 2641 | assert_eq(True & b, True & y) |
| 2642 | assert_eq(True | b, True | y) |
| 2643 | assert_eq(True ^ b, True ^ y) |
| 2644 | assert_eq(2 // b, 2 // y) |
| 2645 | assert_eq(2**b, 2**y) |
| 2646 | assert_eq(2 % b, 2 % y) |
| 2647 | assert_eq(2 > b, 2 > y) |
| 2648 | assert_eq(2 < b, 2 < y) |
| 2649 | assert_eq(2 >= b, 2 >= y) |
| 2650 | assert_eq(2 <= b, 2 <= y) |
| 2651 | assert_eq(2 == b, 2 == y) |
| 2652 | assert_eq(2 != b, 2 != y) |
| 2653 | |