def test_pad_raises(dummy: Tensor) -> None: t = ep.arange(dummy, 120).reshape((2, 3, 4, 5)).float32() ep.pad(t, ((0, 0), (0, 0), (2, 3), (1, 2)), mode="constant") with pytest.raises(ValueError): ep.pad(t, ((0, 0), (2, 3), (1, 2)), mode="constant") with pytest.raises(ValueError): ep.pad( t, ((0, 0), (0, 0, 1, 2), (2, 3), (1, 2)), mode="constant" # type: ignore ) with pytest.raises(ValueError): ep.pad(t, ((0, 0), (0, 0), (2, 3), (1, 2)), mode="foo")
def test_pad(dummy: Tensor, mode: str, value: float) -> Tensor: t = ep.arange(dummy, 120).reshape((2, 3, 4, 5)).float32() return ep.pad(t, ((0, 0), (0, 0), (2, 3), (1, 2)), mode=mode, value=value)