(nx)
| 537 | |
| 538 | |
| 539 | def test_OTResult_LazyTensor(nx): |
| 540 | T, a, b = get_LazyTensor(nx) |
| 541 | |
| 542 | res = ot.utils.OTResult(lazy_plan=T, batch_size=9, backend=nx) |
| 543 | |
| 544 | np.testing.assert_allclose(nx.to_numpy(a), nx.to_numpy(res.marginal_a)) |
| 545 | np.testing.assert_allclose(nx.to_numpy(b), nx.to_numpy(res.marginal_b)) |
| 546 | |
| 547 | |
| 548 | def test_LazyTensor_reduce(nx): |
nothing calls this directly
no test coverage detected