def test_endtoend_twocats(self):
inpfile = os.path.join(TEST_DIRECTORY, 'test_images', 'twocats.png')
im = lab.load_color_image(inpfile)
oim = object_hash(im)
result = lab.seam_carving(im, 100)
lab.save_color_image(result, '/Users/yaxinliu/Downloads/lab1/Seam.png')
def seams_one(images):
for i in images:
inpfile = os.path.join(TEST_DIRECTORY, 'test_images', f'{i}.png')
im = lab.load_color_image(inpfile)
oim = object_hash(im)
result = lab.seam_carving(im, 1)
assert object_hash(im) == oim, 'Be careful not to modify the original image!'
expfile = os.path.join(TEST_DIRECTORY, 'test_results', f'{i}_1seam.png')
compare_color_images(result, lab.load_color_image(expfile))
def seams_endtoend(inp_name, out_name, number):
inpfile = os.path.join(TEST_DIRECTORY, 'test_images', inp_name)
im = lab.load_color_image(inpfile)
oim = object_hash(im)
for i in range(1, number):
result = lab.seam_carving(im, i)
assert object_hash(im) == oim, 'Be careful not to modify the original image!'
expfile = os.path.join(TEST_DIRECTORY, 'test_results', out_name, f'{i:02d}.png')
compare_color_images(result, lab.load_color_image(expfile))
def test_endtoend_smallfrog(self):
inpfile = os.path.join(TEST_DIRECTORY, 'test_images', 'smallfrog.png')
im = lab.load_color_image(inpfile)
oim = object_hash(im)
for i in range(1, 31):
result = lab.seam_carving(im, i)
self.assertEqual(object_hash(im), oim,
'Be careful not to modify the original image!')
expfile = os.path.join(TEST_DIRECTORY, 'test_results',
'seams_smallfrog', f'{i:02d}.png')
self.compare_color_images(result, lab.load_color_image(expfile))