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_idct2_3(self): image = np.array(io.imread('images/sample2048x2048.jpg'), dtype=np.float) idct_image = idct2(image) blocks = skimage.util.view_as_blocks(image, (8, 8)) blocks[:] = idctn(blocks, axes=(2, 3), norm='ortho') self.assertTrue(np.max(np.abs(image - idct_image)) < 1e-10)
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_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_mask_small_3(self): image = np.array(io.imread('images/sample2048x2048.jpg'), dtype=np.float) ulx, lrx, uly, lry = 104, 112, 104, 112 tf_image = idct2(mask(dct2(image), ulx=ulx, lrx=lrx, uly=uly, lry=lry)) image[ulx:lrx, uly:lry] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_mask_horizontal_2(self): image = np.array(io.imread('images/sample1024x1024.jpg'), dtype=np.float) ulx, lrx, uly, lry = 105, 108, 200, 304 tf_image = idct2(mask(dct2(image), ulx=ulx, lrx=lrx, uly=uly, lry=lry)) image[ulx:lrx, uly:lry] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_mask_small_1(self): image = np.array(io.imread('images/sample512x512.jpg'), dtype=np.float) for _ in range(100): original_image = image.copy() ulx = random.randint(0, image.shape[0] - 1) uly = random.randint(0, image.shape[1] - 1) lrx = random.randint(ulx + 1, image.shape[0]) lry = random.randint(uly + 1, image.shape[1]) tf_image = idct2( mask(dct2(image), ulx=ulx, lrx=lrx, uly=uly, lry=lry)) original_image[ulx:lrx, uly:lry] = 0 self.assertTrue(np.max(np.abs(original_image - tf_image)) < 1e-10)
def test_dctidct_3(self): image = np.array(io.imread('images/sample2048x2048.jpg'), dtype=np.float) dct_image = idct2(dct2(image)) self.assertTrue(np.max(np.abs(image - dct_image)) < 1e-10)
def test_rotate90_1(self): image = np.array(io.imread('images/sample512x512.jpg'), dtype=np.float) tf_image = idct2(rotate90(dct2(image))) image = np.rot90(image, k=1) self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_rotate270_3(self): image = np.array(io.imread('images/sample2048x2048.jpg'), dtype=np.float) tf_image = idct2(rotate270(dct2(image))) image = np.rot90(image, k=3) self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_rotate180_2(self): image = np.array(io.imread('images/sample1024x1024.jpg'), dtype=np.float) tf_image = idct2(rotate180(dct2(image))) image = np.rot90(image, k=2) self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_reflecion0_1(self): image = np.array(io.imread('images/sample512x512.jpg'), dtype=np.float) tf_image = idct2(reflection(dct2(image), axis=0)) image = np.flipud(image) self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)
def test_reflecion1_3(self): image = np.array(io.imread('images/sample2048x2048.jpg'), dtype=np.float) tf_image = idct2(reflection(dct2(image), axis=1)) image = np.fliplr(image) 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)
def test_mask_vertical_4(self): image = np.array(io.imread('images/sample512x512.jpg'), dtype=np.float) ulx, lrx, uly, lry = 200, 304, 104, 112 tf_image = idct2(mask(dct2(image), ulx=ulx, lrx=lrx, uly=uly, lry=lry)) image[ulx:lrx, uly:lry] = 0 self.assertTrue(np.max(np.abs(image - tf_image)) < 1e-10)