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,19 +1,19 @@
"""Test image functions."""
import unittest
from pywal import image
from pywal import wal
class TestImage(unittest.TestCase):
"""Test image functions."""
def test_get_img(self):
"""> Validate image file."""
result = image.get_image("tests/test_files/test.jpg")
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 = image.get_image("tests/test_files")
result = wal.get_image("tests/test_files")
self.assertEqual(result, "tests/test_files/test.jpg")

View File

@ -1,7 +1,7 @@
"""Test imagemagick functions."""
import unittest
from pywal import magic
from pywal import wal
class TestGenColors(unittest.TestCase):
@ -9,7 +9,7 @@ class TestGenColors(unittest.TestCase):
def test_gen_colors(self):
"""> Generate a colorscheme."""
result = magic.gen_colors("tests/test_files/test.jpg")
result = wal.create_palette("tests/test_files/test.jpg")
self.assertEqual(result[0], "#0F191A")