pywal/tests/test_colors.py
2017-07-24 00:00:35 +10:00

33 lines
1010 B
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_gen_colors_fail(self):
"""> Generate a colorscheme and fail."""
with self.assertRaises(SystemExit):
colors.get("tests/test_files/test.png")
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")
if __name__ == "__main__":
unittest.main()