From d8376eecc90b659f7b087ddd2ec0f29200ae6141 Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Sun, 23 Jul 2017 11:50:46 +1000 Subject: tests: Update tests. --- 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 da557a7..e0b7902 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, "tests/test_files/test2.jpg") + self.assertEqual(result.endswith(".jpg"), True) if __name__ == "__main__": -- cgit v1.2.3