def test_modify(self):
pic = novice.open(self.small_sample_path)
assert_equal(pic.modified, False)
for p in pic:
if p.x < (pic.width / 2):
p.red /= 2
p.green /= 2
p.blue /= 2
for p in pic:
if p.x < (pic.width / 2):
assert_equal(p.red <= 128, True)
assert_equal(p.green <= 128, True)
assert_equal(p.blue <= 128, True)
s = pic.size
pic.size = (pic.width / 2, pic.height / 2)
assert_equal(pic.size, (int(s[0] / 2), int(s[1] / 2)))
assert_equal(pic.modified, True)
assert_equal(pic.path, None)
with tempfile.NamedTemporaryFile(suffix=".jpg") as tmp:
pic.save(tmp.name)
assert_equal(pic.modified, False)
assert_equal(pic.path, os.path.abspath(tmp.name))
assert_equal(pic.format, "jpeg")
def test_slicing(self):
cut = 40
pic = novice.open(self.sample_path)
rest = pic.width - cut
temp = pic[:cut, :].copy()
pic[:rest, :] = pic[cut:, :]
pic[rest:, :] = temp
def test_modify(self):
pic = novice.open(self.small_sample_path)
assert_equal(pic.modified, False)
for p in pic:
if p.x < (pic.width / 2):
p.red /= 2
p.green /= 2
p.blue /= 2
for p in pic:
if p.x < (pic.width / 2):
assert_equal(p.red <= 128, True)
assert_equal(p.green <= 128, True)
assert_equal(p.blue <= 128, True)
s = pic.size
pic.size = (pic.width / 2, pic.height / 2)
assert_equal(pic.size, (int(s[0] / 2), int(s[1] / 2)))
assert_equal(pic.modified, True)
assert_equal(pic.path, None)
with tempfile.NamedTemporaryFile(suffix=".jpg") as tmp:
pic.save(tmp.name)
assert_equal(pic.modified, False)
assert_equal(pic.path, os.path.abspath(tmp.name))
assert_equal(pic.format, "jpeg")
def test_slicing(self):
cut = 40
pic = novice.open(self.sample_path)
rest = pic.width - cut
temp = pic[:cut, :].copy()
pic[:rest, :] = pic[cut:, :]
pic[rest:, :] = temp
def test_pic_info(self):
pic = novice.open(self.sample_path)
assert_equal(pic.format, "png")
assert_equal(pic.path, os.path.abspath(self.sample_path))
assert_equal(pic.size, (665, 500))
assert_equal(pic.width, 665)
assert_equal(pic.height, 500)
assert_equal(pic.modified, False)
assert_equal(pic.inflation, 1)
num_pixels = sum(1 for p in pic)
assert_equal(num_pixels, pic.width * pic.height)
def test_pic_info(self):
pic = novice.open(self.sample_path)
assert_equal(pic.format, "png")
assert_equal(pic.path, os.path.abspath(self.sample_path))
assert_equal(pic.size, (665, 500))
assert_equal(pic.width, 665)
assert_equal(pic.height, 500)
assert_equal(pic.modified, False)
assert_equal(pic.inflation, 1)
num_pixels = sum(1 for p in pic)
assert_equal(num_pixels, pic.width * pic.height)
def test_indexing(self):
pic = novice.open(self.small_sample_path)
# Slicing
pic[0:5, 0:5] = (0, 0, 0)
for p in pic:
if (p.x < 5) and (p.y < 5):
assert_equal(p.rgb, (0, 0, 0))
assert_equal(p.red, 0)
assert_equal(p.green, 0)
assert_equal(p.blue, 0)
pic[:5, :5] = (255, 255, 255)
for p in pic:
if (p.x < 5) and (p.y < 5):
assert_equal(p.rgb, (255, 255, 255))
assert_equal(p.red, 255)
assert_equal(p.green, 255)
assert_equal(p.blue, 255)
pic[5:pic.width, 5:pic.height] = (255, 0, 255)
for p in pic:
if (p.x >= 5) and (p.y >= 5):
assert_equal(p.rgb, (255, 0, 255))
assert_equal(p.red, 255)
assert_equal(p.green, 0)
assert_equal(p.blue, 255)
pic[5:, 5:] = (0, 0, 255)
for p in pic:
if (p.x >= 5) and (p.y >= 5):
assert_equal(p.rgb, (0, 0, 255))
assert_equal(p.red, 0)
assert_equal(p.green, 0)
assert_equal(p.blue, 255)
# Outside bounds
assert_raises(IndexError, lambda: pic[pic.width, pic.height])
# Negative indexing not supported
assert_raises(IndexError, lambda: pic[-1, -1])
assert_raises(IndexError, lambda: pic[-1:, -1:])
def test_indexing(self):
pic = novice.open(self.small_sample_path)
# Slicing
pic[0:5, 0:5] = (0, 0, 0)
for p in pic:
if (p.x < 5) and (p.y < 5):
assert_equal(p.rgb, (0, 0, 0))
assert_equal(p.red, 0)
assert_equal(p.green, 0)
assert_equal(p.blue, 0)
pic[:5, :5] = (255, 255, 255)
for p in pic:
if (p.x < 5) and (p.y < 5):
assert_equal(p.rgb, (255, 255, 255))
assert_equal(p.red, 255)
assert_equal(p.green, 255)
assert_equal(p.blue, 255)
pic[5:pic.width, 5:pic.height] = (255, 0, 255)
for p in pic:
if (p.x >= 5) and (p.y >= 5):
assert_equal(p.rgb, (255, 0, 255))
assert_equal(p.red, 255)
assert_equal(p.green, 0)
assert_equal(p.blue, 255)
pic[5:, 5:] = (0, 0, 255)
for p in pic:
if (p.x >= 5) and (p.y >= 5):
assert_equal(p.rgb, (0, 0, 255))
assert_equal(p.red, 0)
assert_equal(p.green, 0)
assert_equal(p.blue, 255)
# Outside bounds
assert_raises(IndexError, lambda: pic[pic.width, pic.height])
# Negative indexing not supported
assert_raises(IndexError, lambda: pic[-1, -1])
assert_raises(IndexError, lambda: pic[-1:, -1:])
def test_pixel_iteration(self):
pic = novice.open(self.small_sample_path)
num_pixels = sum(1 for p in pic)
assert_equal(num_pixels, pic.width * pic.height)
def test_pixel_iteration(self):
pic = novice.open(self.small_sample_path)
num_pixels = sum(1 for p in pic)
assert_equal(num_pixels, pic.width * pic.height)