mirror of
https://github.com/dylanaraps/pywal.git
synced 2024-11-25 09:23:08 +01:00
51 lines
1.7 KiB
Python
Executable File
51 lines
1.7 KiB
Python
Executable File
"""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()
|