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.""" """Test image functions."""
import pathlib
import unittest import unittest
from pywal import wal from pywal import wal
DEVNULL = pathlib.Path("/dev/null")
class TestImage(unittest.TestCase): class TestImage(unittest.TestCase):
"""Test image functions.""" """Test image functions."""
def test_get_img(self): def test_get_img(self):
"""> Validate image file.""" """> 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") self.assertEqual(result, "tests/test_files/test.jpg")
def test_get_img_dir(self): def test_get_img_dir(self):
"""> Validate image directory.""" """> Validate image directory."""
result = wal.get_image("tests/test_files") result = wal.get_image("tests/test_files", DEVNULL)
self.assertEqual(result, "tests/test_files/test2.jpg") self.assertEqual(result, "tests/test_files/test.jpg")
if __name__ == "__main__": if __name__ == "__main__":