diff options
| l--------- | tests/test_files/test2.jpg | 1 | ||||
| -rw-r--r-- | tests/test_image.py | 10 |
2 files changed, 7 insertions, 4 deletions
diff --git a/tests/test_files/test2.jpg b/tests/test_files/test2.jpg deleted file mode 120000 index 0e8c5b6..0000000 --- a/tests/test_files/test2.jpg +++ /dev/null @@ -1 +0,0 @@ -test.jpg
\ No newline at end of file diff --git a/tests/test_image.py b/tests/test_image.py index bcf3b4d..b9099f9 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -1,20 +1,24 @@ """Test image functions.""" +import pathlib import unittest from pywal import wal +DEVNULL = pathlib.Path("/dev/null") + + class TestImage(unittest.TestCase): """Test image functions.""" def test_get_img(self): """> Validate image file.""" - result = wal.get_image("tests/test_files/test.jpg") + result = wal.get_image("tests/test_files/test.jpg", DEVNULL) self.assertEqual(result, "tests/test_files/test.jpg") def test_get_img_dir(self): """> Validate image directory.""" - result = wal.get_image("tests/test_files") - self.assertEqual(result, "tests/test_files/test2.jpg") + result = wal.get_image("tests/test_files", DEVNULL) + self.assertEqual(result, "tests/test_files/test.jpg") if __name__ == "__main__": |
