tests: Test image functions (fail)

This commit is contained in:
Dylan Araps 2017-07-23 20:46:29 +10:00
parent af52eb32dc
commit 7eca144a8f

View File

@ -16,6 +16,16 @@ class TestImage(unittest.TestCase):
result = image.get("tests/test_files")
self.assertEqual(result.endswith(".jpg"), True)
def test_get_img_fail(self):
"""> Validate image file. (fail)"""
with self.assertRaises(SystemExit):
image.get("tests/test_files/test_fail.jpg")
def test_get_img_dir_fail(self):
"""> Validate image directory. (fail)"""
with self.assertRaises(SystemExit):
image.get("tests")
if __name__ == "__main__":
unittest.main()