diff --git a/tests/integration/full/full_test_deposition.py b/tests/integration/full/full_test_deposition.py index 8f06e5ad6a31265c8e82790d59545a990062f32a..1291706402ca9714ce7b136a43c0e453b25af6d5 100644 --- a/tests/integration/full/full_test_deposition.py +++ b/tests/integration/full/full_test_deposition.py @@ -1,5 +1,6 @@ import argparse import copy +import glob import os import sys import unittest @@ -55,13 +56,13 @@ class TestFullDeposition(unittest.TestCase): east_africa_image_path = os.path.join(TestFullDeposition.TEST_OUT_PATH, TestFullDeposition.TEST_JOB_DIR, "plotting", "eastafrica", "images", "Weekly", - "deposition_eastafrica_leaf_rust_total_202301260000_202302020000_map.png") + "deposition_eastafrica_leaf_rust_total_*_map.png") ethiopia_image_path = os.path.join(TestFullDeposition.TEST_OUT_PATH, TestFullDeposition.TEST_JOB_DIR, "plotting", "ethiopia", "images", "Weekly", - "deposition_ethiopia_leaf_rust_total_202301260000_202302020000_map.png") + "deposition_ethiopia_leaf_rust_total_*_map.png") - ea_file_exists: bool = os.path.isfile(east_africa_image_path) - eth_file_exists: bool = os.path.isfile(ethiopia_image_path) + ea_file_exists: bool = len(glob.glob(east_africa_image_path)) > 1 + eth_file_exists: bool = len(glob.glob(ethiopia_image_path)) > 1 self.assertTrue(ea_file_exists) self.assertTrue(eth_file_exists)