(self)
| 110 | |
| 111 | class TestReduce: |
| 112 | def test_sum(self) -> None: |
| 113 | buf = rt.asarray(np.arange(100, dtype=np.float64)) |
| 114 | total = rt.go(kernel.sum(buf)).result().scalar() |
| 115 | assert total == 4950.0 |
| 116 | |
| 117 | def test_mean(self) -> None: |
| 118 | buf = rt.asarray(np.arange(100, dtype=np.float64)) |