def test_bad_clipping_range(image, header):
# Case 1: x completely out of range
with pytest.raises(IndexError):
flipped = clipnflip(image, header, xrange=(100, 150))
# Case 2: x upper limit out of range
with pytest.raises(IndexError):
flipped = clipnflip(image, header, xrange=(5, 150))
# Case 3: x limits inverse
with pytest.raises(IndexError):
flipped = clipnflip(image, header, xrange=(10, 0))
# Case 4: x limits only one column
with pytest.raises(IndexError):
flipped = clipnflip(image, header, xrange=(1, 1))
# Case 1: y completely out of range
with pytest.raises(IndexError):
flipped = clipnflip(image, header, yrange=(100, 150))
# Case 2: y upper limit out of range
with pytest.raises(IndexError):
flipped = clipnflip(image, header, yrange=(5, 150))
# Case 3: y limits inverse
with pytest.raises(IndexError):
flipped = clipnflip(image, header, yrange=(10, 0))
# Case 4: y limits only one column
with pytest.raises(IndexError):
flipped = clipnflip(image, header, yrange=(1, 1))
def test_only_clip(image, header):
# clip x direction
flipped = clipnflip(image, header, xrange=(5, 8))
assert isinstance(flipped, np.ndarray)
assert flipped.shape[0] == 20
assert flipped.shape[1] == 3
assert np.all(flipped == image[:, 5:8])
# clip y direction
flipped = clipnflip(image, header, yrange=(10, 15))
assert isinstance(flipped, np.ndarray)
assert flipped.shape[0] == 5
assert flipped.shape[1] == 10
assert np.all(flipped == image[10:15, :])
def test_nochange(image, header):
# This should change nothing
flipped = clipnflip(image, header)
assert isinstance(flipped, np.ndarray)
assert flipped.shape[0] == image.shape[0]
assert flipped.shape[1] == image.shape[1]
assert np.all(flipped == image)
def test_multidimensional(image, header):
image = image[None, None, ...]
flipped = clipnflip(image, header)
assert isinstance(flipped, np.ndarray)
assert flipped.shape[0] == image.shape[-2]
assert flipped.shape[1] == image.shape[-1]
assert np.all(flipped == image[0, 0])
def test_only_rotate(image, header):
for orient in [0, 1, 2, 3]:
flipped = clipnflip(image, header, orientation=orient)
compare = np.rot90(image, -1 * orient)
assert isinstance(flipped, np.ndarray)
assert flipped.shape[0] == compare.shape[0]
assert flipped.shape[1] == compare.shape[1]
assert np.all(flipped == compare)