tests: Cover random image.

1 files changed, 4 insertions, 4 deletions

diff --git a/tests/test_image.py b/tests/test_image.py
index b9099f9..95a357e 100644
--- a/tests/test_image.py
+++ b/tests/test_image.py
@@ -5,20 +5,20 @@ import unittest
 from pywal import wal
 
 
-DEVNULL = pathlib.Path("/dev/null")
+CACHE_DIR = pathlib.Path("/tmp/wal")
 
 
 class TestImage(unittest.TestCase):
     """Test image functions."""
     def test_get_img(self):
         """> Validate image file."""
-        result = wal.get_image("tests/test_files/test.jpg", DEVNULL)
+        result = wal.get_image("tests/test_files/test.jpg", CACHE_DIR)
         self.assertEqual(result, "tests/test_files/test.jpg")
 
     def test_get_img_dir(self):
         """> Validate image directory."""
-        result = wal.get_image("tests/test_files", DEVNULL)
-        self.assertEqual(result, "tests/test_files/test.jpg")
+        result = wal.get_image("tests/test_files", CACHE_DIR)
+        self.assertEqual(result, "tests/test_files/test2.jpg")
 
 
 if __name__ == "__main__":