tests: Fix tests

This commit is contained in:
Dylan Araps 2017-07-22 10:26:36 +10:00
parent ccad5150e6
commit 3dd1fc9039
2 changed files with 7 additions and 4 deletions

View File

@ -1 +0,0 @@
test.jpg

View File

@ -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__":