Esempio n. 1
0
    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")
Esempio n. 2
0
 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
Esempio n. 3
0
    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")
Esempio n. 4
0
 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
Esempio n. 5
0
    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)
Esempio n. 6
0
    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)
Esempio n. 7
0
    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:])
Esempio n. 8
0
    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:])
Esempio n. 9
0
 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)
Esempio n. 10
0
 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)