diff --git a/tests/test_flip_image.py b/tests/test_flip_image.py new file mode 100644 index 0000000000000000000000000000000000000000..bfce6b750f378d9667efad47675a59dc77c04cb5 --- /dev/null +++ b/tests/test_flip_image.py @@ -0,0 +1,21 @@ +import unittest + +import numpy as np + +from csst_proto.top_level_interface import flip_image, read_test_image + + +class FlipImageTestCase(unittest.TestCase): + def test_flip_image(self): + # flip test image + self.assertTrue( + np.all(flip_image(read_test_image()) == np.array([[4, 3], [2, 1]])) + ) + + # the code fails for 1D array + with self.assertRaises(AssertionError): + flip_image(np.array([1, 2, 3, 4])) + + +if __name__ == "__main__": + unittest.main()