aboutsummaryrefslogtreecommitdiff
path: root/tests/test_image.py
blob: 95a357ef96423a10ce51c3984dbe565a19bb595a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
"""Test image functions."""
import pathlib
import unittest

from pywal import wal


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", 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", CACHE_DIR)
        self.assertEqual(result, "tests/test_files/test2.jpg")


if __name__ == "__main__":
    unittest.main()