From e878957ea8aaaafa28347d7221b8da0817cd903c Mon Sep 17 00:00:00 2001 From: lb584 <lb584@cam.ac.uk> Date: Tue, 15 Nov 2022 11:48:25 +0000 Subject: [PATCH] updating advisory test --- tests/integration/test_advisory.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/tests/integration/test_advisory.py b/tests/integration/test_advisory.py index cfa8d72..a5bde54 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() -- GitLab