README Examples §4: sum, mean, min_reduce, max_reduce.
()
| 52 | |
| 53 | |
| 54 | def test_example4_reductions() -> None: |
| 55 | """README Examples §4: sum, mean, min_reduce, max_reduce.""" |
| 56 | buf = rt.asarray(np.arange(100, dtype=np.float64)) |
| 57 | |
| 58 | assert rt.go(kernel.sum(buf)).result().scalar() == 4950.0 |
| 59 | assert rt.go(kernel.mean(buf)).result().scalar() == 49.5 |
| 60 | assert rt.go(kernel.min_reduce(buf)).result().scalar() == 0.0 |
| 61 | assert rt.go(kernel.max_reduce(buf)).result().scalar() == 99.0 |
| 62 | |
| 63 | |
| 64 | def test_example5_relu_where() -> None: |
nothing calls this directly
no test coverage detected