From 5f6de5efe82764c141019e47b488e5def62ca4c2 Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Sun, 23 Jul 2017 21:29:38 +1000 Subject: tests: Fix errors --- tests/test_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/test_image.py') diff --git a/tests/test_image.py b/tests/test_image.py index 1cf7ec6..195951d 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -14,7 +14,7 @@ class TestImage(unittest.TestCase): def test_get_img_dir(self): """> Validate image directory.""" result = image.get("tests/test_files") - self.assertEqual(result.endswith(".jpg"), True) + self.assertEqual(result.endswith((".jpg", ".png")), True) def test_get_img_fail(self): """> Validate image file. (fail)""" -- cgit v1.2.3