tests: Update tests.

This commit is contained in:
Dylan Araps
2017-07-23 11:50:46 +10:00
parent 7e5723d7c1
commit d8376eecc9
2 changed files with 6 additions and 6 deletions

View File

@@ -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__":