def test_translationnegative1_3(self): image = np.array(io.imread('images/sample2048x2048.jpg'), dtype=np.float) tf_image = idct2(translation(dct2(image), -100, axis=1)) image[:, :-100] = image[:, 100:] image[:, -100:] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_translationpositive1_2(self): image = np.array(io.imread('images/sample1024x1024.jpg'), dtype=np.float) tf_image = idct2(translation(dct2(image), 100, axis=1)) image[:, 100:] = image[:, :-100] image[:, :100] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_translationnegative0_2(self): image = np.array(io.imread('images/sample1024x1024.jpg'), dtype=np.float) tf_image = idct2(translation(dct2(image), -100, axis=0)) image[:-100, :] = image[100:, :] image[-100:, :] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_translationpositive0_1(self): image = np.array(io.imread('images/sample512x512.jpg'), dtype=np.float) tf_image = idct2(translation(dct2(image), 100, axis=0)) image[100:, :] = image[:-100, :] image[:100, :] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_translationzero_2(self): image = np.array(io.imread('images/sample512x512.jpg'), dtype=np.float) tf_image = idct2(translation(dct2(image), image.shape[1] * 2, axis=0)) image[:] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)