(
self,
target: NamedArray[Any, np.dtype[np.float32]],
broadcast_dims: Mapping[_Dim, int],
expected_ndim: int,
)
| 565 | ], |
| 566 | ) |
| 567 | def test_broadcast_to( |
| 568 | self, |
| 569 | target: NamedArray[Any, np.dtype[np.float32]], |
| 570 | broadcast_dims: Mapping[_Dim, int], |
| 571 | expected_ndim: int, |
| 572 | ) -> None: |
| 573 | expand_dims = set(broadcast_dims.keys()) - set(target.dims) |
| 574 | # loop over expand_dims and call .expand_dims(dim=dim) in a loop |
| 575 | for dim in expand_dims: |
| 576 | target = target.expand_dims(dim=dim) |
| 577 | result = target.broadcast_to(broadcast_dims) |
| 578 | assert result.ndim == expected_ndim |
| 579 | assert result.sizes == broadcast_dims |
| 580 | |
| 581 | def test_broadcast_to_errors( |
| 582 | self, target: NamedArray[Any, np.dtype[np.float32]] |
nothing calls this directly
no test coverage detected