blob: 54e72360f373ee166f6312c83284739c892e271e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
"""Test imagemagick functions."""
import unittest
import unittest.mock
import io
import pathlib
from pywal import colors
from pywal import util
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_cache(self):
"""> Test importing a cached scheme."""
# Create the file to import.
cache_dir = pathlib.Path("/tmp/wal2/")
util.save_file("true", cache_dir / "schemes"
/ "tests_test_files_test_jpg.json")
# Since this function just prints a message we redirect
# it's output so that we can read it.
message = "colors: Found cached colorscheme."
with unittest.mock.patch('sys.stdout', new=io.StringIO()) as fake_out:
colors.get("tests/test_files/test.jpg", cache_dir)
self.assertEqual(fake_out.getvalue().strip(), message)
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()
|
