diff --git a/tests/integration/test_advisory.py b/tests/integration/test_advisory.py index cfa8d7240f0e3f48e17cbf664ebf6e23e1c628ac..a5bde542faf57542efa0d0d7517ad645105dbdaf 100644 --- a/tests/integration/test_advisory.py +++ b/tests/integration/test_advisory.py @@ -1,4 +1,5 @@ import copy +import glob import os import unittest @@ -67,6 +68,29 @@ class TestAdvisory(unittest.TestCase): self.assertTrue(ea_file_exists) self.assertTrue(eth_file_exists) + def test_standard_run_input_images_produced(self): + + east_africa_image_wildcard = os.path.join(TestAdvisory.TEST_OUT_PATH, TestAdvisory.TEST_JOB_DIR, + "images", + "*eastafrica*.png") + ethiopia_image_wildcard = os.path.join(TestAdvisory.TEST_OUT_PATH, TestAdvisory.TEST_JOB_DIR, + "images", + "*ethiopia*.png") + + africa_image_count: int = len(glob.glob(east_africa_image_wildcard)) + ethiopia_image_count: int = len(glob.glob(ethiopia_image_wildcard)) + self.assertEqual(3, africa_image_count) + self.assertEqual(3, ethiopia_image_count) + + def test_standard_run_input_shapefiles_produced(self): + + east_africa_image_wildcard = os.path.join(TestAdvisory.TEST_OUT_PATH, TestAdvisory.TEST_JOB_DIR, + "images", + "*eastafrica*.shp") + + africa_image_count: int = len(glob.glob(east_africa_image_wildcard)) + self.assertEqual(3, africa_image_count) + if __name__ == '__main__': unittest.main()