test_colorific.py

# -*- coding: utf-8 -*-
#
#  test_colorific.py
#  palette-detect
#

import os
import unittest
import itertools

import colorific
from colorific import palette

CORE_COLORS = [
    '#000000',  # black
    '#0000ff',  # blue
    '#00ff00',  # green
    '#ff0000',  # red
    '#ffffff',  # white
]


class ExtractionTest(unittest.TestCase):
    def setUp(self):
        self.filename = os.path.join(os.path.dirname(__file__),
                                     'nasa_ares_logo.png')

    def test_extraction(self):
        expected = [(0, 101, 185),
                    (187, 214, 236),
                    (255, 0, 0)]
        p = palette.extract_colors(self.filename)
        found = [c.value for c in p.colors]
        self.assertEquals(found, expected)


class ConversionTest(unittest.TestCase):
    def setUp(self):
        self.pairs = [
            ((0, 0, 0), '#000000'),
            ((255, 255, 255), '#ffffff'),
            ((255, 0, 0), '#ff0000'),
            ((0, 255, 0), '#00ff00'),
            ((0, 0, 255), '#0000ff'),
        ]

    def test_hex_to_rgb(self):
        for rgb, hexval in self.pairs:
            self.assertEqual(colorific.hex_to_rgb(hexval), rgb)

    def test_rgb_to_hex(self):
        for rgb, hexval in self.pairs:
            self.assertEqual(colorific.rgb_to_hex(rgb), hexval)


class VisualDistanceTest(unittest.TestCase):
    def test_core_colors(self):
        for c1, c2 in itertools.combinations(CORE_COLORS, 2):
            assert not self.similar(c1, c2)

    def test_apparent_mistakes(self):
        mistakes = [
            ('#f1f1f1', '#f2f2f2'),
            ('#f2f2f2', '#f3f3f3'),
            ('#fafafa', '#fbfbfb'),
            ('#7c7c7c', '#7d7d7d'),
            ('#29abe1', '#29abe2'),
        ]

        for c1, c2 in mistakes:
            assert self.similar(c1, c2)

    def distance(self, c1, c2):
        return colorific.distance(
            colorific.hex_to_rgb(c1),
            colorific.hex_to_rgb(c2),
        )

    def similar(self, c1, c2):
        return self.distance(c1, c2) < colorific.MIN_DISTANCE


def suite():
    return unittest.TestSuite((
        unittest.makeSuite(ExtractionTest),
        unittest.makeSuite(ConversionTest),
        unittest.makeSuite(VisualDistanceTest),
    ))

if __name__ == '__main__':
    unittest.TextTestRunner(verbosity=1).run(suite())