def test_pad_boxsum_non_zero(array_size_3):
"""Test padding with a number other than the default of 0."""
padded = pad_boxsum(array_size_3, 3, mode="constant", constant_values=2)
expected = 2 * np.ones((6, 6), dtype=np.int32)
expected[2:5, 2:5] = array_size_3
np.testing.assert_array_equal(padded, expected)
def test_pad_boxsum(array_size_3):
"""Test that padded array consists of input array surrounded by border of zeros."""
padded = pad_boxsum(array_size_3, 3, mode="constant")
expected = np.zeros((6, 6), dtype=np.int32)
expected[2:5, 2:5] = array_size_3
np.testing.assert_array_equal(padded, expected)