def test_load(self):
""" Test load. """
img = CImg()
self.assertRaises(RuntimeError, img.load, 'notexistent.jpg')
img.load(get_test_image())
self._check_image_dimensions(img)
def test_acos(self):
arr = -np.ones((10, 5))
img = CImg(arr)
self.assertTrue(np.allclose(np.arccos(arr), img.acos().asarray()))
def test_atan(self):
arr = np.random.randn(10, 5)
img = CImg(arr)
self.assertTrue(np.allclose(np.arctan(arr), img.atan().asarray()))
def test_fill(self):
""" Test fill. """
img = CImg((2, 2))
img.fill(42)
img_expected = CImg(np.ones((2, 2)) * 42)
self.assertEqual(img, img_expected)
def test_min_max(self):
""" Test min_max. """
img = CImg(np.array([[2, -1], [0, 4]]))
self.assertEqual(img.min_max(), (-1, 4))
def test_load_png(self):
""" Test loading a PNG file. """
img = CImg()
img.load_png(get_test_image('png'))
self._check_image_dimensions(img)
def test_round(self):
""" Test round. """
img = CImg(np.array([2.3, -1.7, 1.5, -0.1]))
img.round()
img_expected = CImg(np.array([2, -2, 2, 0]))
self.assertEqual(img, img_expected)
def test_dot(self):
""" Test dot. """
img1 = CImg(np.array([[2, -5], [0, 3]]))
img2 = CImg(np.array([[2, -3], [0, 3]]))
self.assertEqual(img1.dot(img2), 28)
def test_log10(self):
exponents = np.linspace(-10, 10, 11)
arr = (10 * np.ones((1, 11)))**exponents
img = CImg(arr)
self.assertTrue(np.allclose(exponents, img.log10().asarray()))
def test_mse(self):
""" Test MSE. """
img1 = CImg(np.array([[2, -5], [0, 3]]))
img2 = CImg(np.array([[2, -3], [0, 3]]))
self.assertEqual(img1.mse(img2), 1.0)
def test_magnitude(self):
""" Test magnitude. """
img = CImg(np.array([[2, -5], [0, 3]]))
self.assertEqual(img.magnitude(cimg.L1_NORM), 10)
def test_variance_mean(self):
""" Test variance + mean. """
img = CImg(np.array([[2, -5], [0, 3]]))
self.assertEqual(img.variance_mean(), (38.0 / 3.0, 0))
def test_variance(self):
""" Test variance. """
img = CImg(np.array([[2, -5], [0, 3]]))
self.assertEqual(img.variance(), 38.0 / 3.0)
def test_max_min(self):
""" Test max_min. """
img = CImg(np.array([[2, -1], [0, 4]]))
self.assertEqual(img.max_min(), (4, -1))
def test_load_bmp(self):
""" Test loading a BMP file. """
img = CImg()
img.load_bmp(get_test_image('bmp'))
self._check_image_dimensions(img)
def test_sinc(self):
arr = np.linspace(-10, 10, 100)
img = CImg(arr)
self.assertTrue(np.allclose(np.sinc(arr / np.pi),
img.sinc().asarray()))
def test_load_jpeg(self):
""" Test loading a JPEG file. """
img = CImg()
img.load_jpeg(get_test_image('jpg'))
self._check_image_dimensions(img)
def test_sqr(self):
arr = np.random.randn(10, 5)
img = CImg(arr)
self.assertTrue(np.allclose(arr * arr, img.sqr().asarray()))
def test_load_tiff(self):
""" Test loading a TIFF file. """
img = CImg()
img.load_tiff(get_test_image('tiff'))
self._check_image_dimensions(img)
def test_tan(self):
arr = np.linspace(-0.5, 0.5, 100)
img = CImg(arr)
self.assertTrue(np.allclose(np.tan(arr), img.tan().asarray()))
def test_noise(self):
""" Test noise. """
img = CImg(np.zeros((5, 5)))
img.noise(2)
img_not_expected = CImg(np.zeros((5, 5)))
self.assertNotEqual(img, img_not_expected)
def test_tanh(self):
arr = np.random.randn(10, 5)
img = CImg(arr)
self.assertTrue(np.allclose(np.tanh(arr), img.tanh().asarray()))
def test_quantize(self):
""" Test quantize. """
img = CImg(np.array([0, 1, 2, 3, 4]))
img.quantize(2, keep_range=True)
img_expected = CImg(np.array([0, 0, 2, 2, 2]))
self.assertEqual(img, img_expected)
def test_sqrt(self):
arr = np.ones((10, 5))
img = CImg(arr)
self.assertTrue(np.allclose(arr, img.sqrt().asarray()))