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)
