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