pywal/tests/test_colors.py
2018-07-06 08:36:29 +10:00

33 lines
1 KiB
Python
Executable file

"""Test imagemagick functions."""
import unittest
from pywal import colors
class TestGenColors(unittest.TestCase):
"""Test the gen_colors functions."""
def test_gen_colors(self):
"""> Generate a colorscheme."""
result = colors.get("tests/test_files/test.jpg")
self.assertEqual(len(result["colors"]["color0"]), 7)
def test_color_import(self):
"""> Read colors from a file."""
result = colors.file("tests/test_files/test_file.json")
self.assertEqual(result["colors"]["color0"], "#1F211E")
def test_color_import_no_wallpaper(self):
"""> Read colors from a file without a wallpaper."""
result = colors.file("tests/test_files/test_file2.json")
self.assertEqual(result["wallpaper"], "None")
def test_color_import_no_alpha(self):
"""> Read colors from a file without an alpha."""
result = colors.file("tests/test_files/test_file2.json")
self.assertEqual(result["alpha"], "100")
if __name__ == "__main__":
unittest.main()