tests: Fix tests.

This commit is contained in:
Dylan Araps
2017-07-20 23:39:19 +10:00
parent ced562e4df
commit 4fd592b7d0
2 changed files with 5 additions and 5 deletions

View File

@ -1,21 +0,0 @@
"""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()