diff options
Diffstat (limited to 'tests/test_colors.py')
| -rwxr-xr-x | tests/test_colors.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/test_colors.py b/tests/test_colors.py index 52957fd..5765370 100755 --- a/tests/test_colors.py +++ b/tests/test_colors.py @@ -2,16 +2,20 @@ import unittest import unittest.mock import io +import pathlib from pywal import colors +CACHE_DIR = pathlib.Path("/tmp/wal") + + 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") + result = colors.get("tests/test_files/test.jpg", CACHE_DIR) self.assertEqual(len(result["colors"]["color0"]), 7) def test_gen_colors_fail(self): @@ -25,7 +29,7 @@ class TestGenColors(unittest.TestCase): # 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") + colors.get("tests/test_files/test.jpg", CACHE_DIR) self.assertEqual(fake_out.getvalue().strip(), message) def test_color_import(self): |