(self)
| 48 | np.testing.assert_array_equal(out, [-1.0, 2.0, 0.0]) |
| 49 | |
| 50 | def test_subtraction(self) -> None: |
| 51 | x, y = kernel.args("x", "y") |
| 52 | sub = kernel.elementwise(x - y) |
| 53 | bx = rt.asarray(np.array([5.0, 3.0])) |
| 54 | by = rt.asarray(np.array([2.0, 1.0])) |
| 55 | out = rt.go(kernel.map(sub, x=bx, y=by)).result().numpy() |
| 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") |