From 4fd592b7d0a5fcf6674d07f34e6d92051634fc6c Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Thu, 20 Jul 2017 23:39:19 +1000 Subject: tests: Fix tests. --- tests/test_image.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 tests/test_image.py (limited to 'tests/test_image.py') diff --git a/tests/test_image.py b/tests/test_image.py new file mode 100644 index 0000000..7dacecb --- /dev/null +++ b/tests/test_image.py @@ -0,0 +1,21 @@ +"""Test image functions.""" +import unittest + +from pywal import 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") + 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/test.jpg") + + +if __name__ == "__main__": + unittest.main() -- cgit v1.2.3