From 5f6de5efe82764c141019e47b488e5def62ca4c2 Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Sun, 23 Jul 2017 21:29:38 +1000 Subject: tests: Fix errors --- tests/test_colors.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'tests/test_colors.py') 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): -- cgit v1.2.3