def test_untile_non_overlapping_patches(kernel_size: int, stride: int, image_size: torch.Size) -> None:
    """Non-Overlapping Tiling/Untiling should return the same image size."""
    tiler = Tiler(tile_size=kernel_size, stride=stride)
    image = torch.rand(image_size)
    tiles = tiler.tile(image)

    untiled_image = tiler.untile(tiles)
    assert untiled_image.shape == torch.Size(image_size)
Non-Overlapping Tiling/Untiling should return the same image size.
tests/unit/data/utils/test_tiler.py