From 4fd592b7d0a5fcf6674d07f34e6d92051634fc6c Mon Sep 17 00:00:00 2001
From: Dylan Araps <dylan.araps@gmail.com>
Date: Thu, 20 Jul 2017 23:39:19 +1000
Subject: tests: Fix tests.

---
 tests/test_colors.py | 21 ---------------------
 tests/test_image.py  | 21 +++++++++++++++++++++
 tests/test_magic.py  |  4 ++--
 3 files changed, 23 insertions(+), 23 deletions(-)
 delete mode 100644 tests/test_colors.py
 create mode 100644 tests/test_image.py

diff --git a/tests/test_colors.py b/tests/test_colors.py
deleted file mode 100644
index 077600e..0000000
--- a/tests/test_colors.py
+++ /dev/null
@@ -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()
diff --git a/tests/test_image.py b/tests/test_image.py
new file mode 100644
index 0000000..7dacecb
--- /dev/null
+++ b/tests/test_image.py
@@ -0,0 +1,21 @@
+"""Test image functions."""
+import unittest
+
+from pywal import wal
+
+
+class TestImage(unittest.TestCase):
+    """Test image functions."""
+    def test_get_img(self):
+        """> Validate image file."""
+        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 = wal.get_image("tests/test_files")
+        self.assertEqual(result, "tests/test_files/test.jpg")
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/tests/test_magic.py b/tests/test_magic.py
index 673420c..852f8b3 100755
--- a/tests/test_magic.py
+++ b/tests/test_magic.py
@@ -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")
 
 
-- 
cgit v1.2.3