From 930c3c8e40ca5c563d0414e5057b23ad27aa7cf3 Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Fri, 30 Jun 2017 10:10:07 +1000 Subject: tests: Update tests. --- tests/test_colors.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 tests/test_colors.py (limited to 'tests/test_colors.py') diff --git a/tests/test_colors.py b/tests/test_colors.py new file mode 100644 index 0000000..077600e --- /dev/null +++ b/tests/test_colors.py @@ -0,0 +1,21 @@ +"""Test image functions.""" +import unittest + +from pywal import image + + +class TestImage(unittest.TestCase): + """Test image functions.""" + def test_get_img(self): + """> Validate image file.""" + result = image.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 = image.get_image("tests/test_files") + self.assertEqual(result, "tests/test_files/test.jpg") + + +if __name__ == "__main__": + unittest.main() -- cgit v1.2.3