(self)
| 56 | np.testing.assert_array_equal(out, [3.0, 2.0]) |
| 57 | |
| 58 | def test_division(self) -> None: |
| 59 | x, y = kernel.args("x", "y") |
| 60 | div = kernel.elementwise(x / y) |
| 61 | bx = rt.asarray(np.array([6.0, 9.0])) |
| 62 | by = rt.asarray(np.array([2.0, 3.0])) |
| 63 | out = rt.go(kernel.map(div, x=bx, y=by)).result().numpy() |
| 64 | np.testing.assert_array_equal(out, [3.0, 3.0]) |
| 65 | |
| 66 | def test_reverse_ops(self) -> None: |
| 67 | x = kernel.arg("x") |