tests: Test image cache
"""Test imagemagick functions."""
import unittest
+import unittest.mock
+import io
from pywal import colors
@@ -17,6 +19,15 @@ class TestGenColors(unittest.TestCase):
with self.assertRaises(SystemExit):
colors.get("tests/test_files/test.png")
+ def test_color_cache(self):
+ """> Test importing a cached scheme."""
+ # 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")
+ self.assertEqual(fake_out.getvalue().strip(), message)
+
if __name__ == "__main__":
unittest.main()
