From 6fd5efe1424393f8100c7de8c65fc568c4759287 Mon Sep 17 00:00:00 2001 From: lb584 <lb584@cam.ac.uk> Date: Thu, 21 Dec 2023 14:02:45 +0000 Subject: [PATCH] more work wiring in nc files as output --- coordinator/EnvSuitPipeline.py | 4 ++-- coordinator/ProcessorEnvironment.py | 2 +- tests/integration/full/full_test_epi.py | 6 +++--- tests/integration/partial/integration_test_utils.py | 6 +++--- tests/integration/partial/test_advisory.py | 2 +- tests/integration/partial/test_deposition.py | 2 +- tests/integration/partial/test_env_suit.py | 2 +- tests/integration/partial/test_epi.py | 2 +- tests/integration/partial/test_survey.py | 2 +- tests/integration/test_suites/depo_test_suite.py | 4 ++-- tests/integration/test_suites/env_suit_test_suite.py | 4 ++-- 11 files changed, 18 insertions(+), 18 deletions(-) diff --git a/coordinator/EnvSuitPipeline.py b/coordinator/EnvSuitPipeline.py index 20e570c..5fb07ec 100644 --- a/coordinator/EnvSuitPipeline.py +++ b/coordinator/EnvSuitPipeline.py @@ -261,9 +261,9 @@ def run_pipeline(pipeline_config, region, dateString, extracted = False, prevent # run_merger(config, sys_config, processor_name) - resultFile = envSuitPath + 'RIE.tif' + resultFile = envSuitPath + 'RIE.nc' strain_outPath = os.path.join(region_outPath,strain) - strain_outFile = strain_outPath + '/RIE_value.tif' + strain_outFile = strain_outPath + '/RIE_value.nc' # Check results dimension # result = pd.read_csv(resultFile) diff --git a/coordinator/ProcessorEnvironment.py b/coordinator/ProcessorEnvironment.py index d3b70db..5780126 100644 --- a/coordinator/ProcessorEnvironment.py +++ b/coordinator/ProcessorEnvironment.py @@ -180,7 +180,7 @@ class ProcessorEnvironment(Processor): for region in subregions: output_dir = f"{jobPath}/plotting/{region.lower()}" - csv_template_dir = input_dir + "/{DISEASE_DIR}/RIE_value.tif" + csv_template_dir = input_dir + "/{DISEASE_DIR}/RIE_value.nc" Path(output_dir).mkdir(parents=True, exist_ok=True) diff --git a/tests/integration/full/full_test_epi.py b/tests/integration/full/full_test_epi.py index 33fb8ed..1c96b33 100644 --- a/tests/integration/full/full_test_epi.py +++ b/tests/integration/full/full_test_epi.py @@ -20,10 +20,10 @@ class FullTestEpi(BaseEpiTestSuite.EpiTestSuite): self.set_expected_values() - if IntegrationTestUtils.TEST_OUT_PATH is None or not os.path.isdir(IntegrationTestUtils.TEST_OUT_PATH): - # if True: + # if IntegrationTestUtils.TEST_OUT_PATH is None or not os.path.isdir(IntegrationTestUtils.TEST_OUT_PATH): + if True: FullTestEpi.write_temp_run_config_files() - FullTestEpi.run_dependent_pipelines() + # FullTestEpi.run_dependent_pipelines() FullTestEpi.run_epi_pipeline() else: print(f"output in {IntegrationTestUtils.TEST_OUT_PATH} already written, skipping rerun") diff --git a/tests/integration/partial/integration_test_utils.py b/tests/integration/partial/integration_test_utils.py index 9903274..2003841 100644 --- a/tests/integration/partial/integration_test_utils.py +++ b/tests/integration/partial/integration_test_utils.py @@ -83,8 +83,8 @@ class IntegrationTestUtils: IntegrationTestUtils.DEFAULT_CONFIG_FILE_PATH = _config_file nowstring: str = IntegrationTestUtils.get_now_string() - prefix: str = f"temp_{test_prefix}_" + nowstring - # prefix: str = f"temp_{test_prefix}" + # prefix: str = f"temp_{test_prefix}_" + nowstring + prefix: str = f"temp_{test_prefix}" IntegrationTestUtils.TEST_OUT_PATH = _outdir + prefix + os.sep IntegrationTestUtils.EMAIL_CRED_PATH = _email_cred_path @@ -264,6 +264,6 @@ class IntegrationTestUtils: result = custom_run_date else: print("default") - result = "20230126" + result = "20231127" return result diff --git a/tests/integration/partial/test_advisory.py b/tests/integration/partial/test_advisory.py index d8558cd..bdfe984 100644 --- a/tests/integration/partial/test_advisory.py +++ b/tests/integration/partial/test_advisory.py @@ -18,7 +18,7 @@ class TestAdvisory(BaseAdvisoryTestSuite.AdvisoryTestSuite): IntegrationTestUtils.TEST_OUT_PATH = None def set_expected_values(self): - IntegrationTestUtils.TEST_START_DATE = '20221001' + IntegrationTestUtils.TEST_START_DATE = '20231127' IntegrationTestUtils.TEST_JOB_DIR = "ADVISORY_" + IntegrationTestUtils.TEST_START_DATE self.EXPECTED_EA_IMAGE_COUNT = 6 self.EXPECTED_ETH_IMAGE_COUNT = 6 diff --git a/tests/integration/partial/test_deposition.py b/tests/integration/partial/test_deposition.py index 96268cb..8e665e4 100644 --- a/tests/integration/partial/test_deposition.py +++ b/tests/integration/partial/test_deposition.py @@ -19,7 +19,7 @@ class TestDeposition(BaseDepoTestSuite.DepoTestSuite): def set_expected_values(self): - IntegrationTestUtils.TEST_START_DATE = "20221001" + IntegrationTestUtils.TEST_START_DATE = "20231127" IntegrationTestUtils.TEST_JOB_DIR = "DEPOSITION_" + IntegrationTestUtils.TEST_START_DATE self.EA_CSV_COUNT = 9 diff --git a/tests/integration/partial/test_env_suit.py b/tests/integration/partial/test_env_suit.py index 843556a..55c7d82 100644 --- a/tests/integration/partial/test_env_suit.py +++ b/tests/integration/partial/test_env_suit.py @@ -18,7 +18,7 @@ class TestEnvSuit(BaseEnvSuitTestSuite.EnvSuitTestSuite): IntegrationTestUtils.TEST_OUT_PATH = None def set_expected_values(self): - IntegrationTestUtils.TEST_START_DATE = "20221001" + IntegrationTestUtils.TEST_START_DATE = "20231127" IntegrationTestUtils.TEST_JOB_DIR = "ENVIRONMENT_2.0_" + IntegrationTestUtils.TEST_START_DATE self.EA_CSV_COUNT = 18 diff --git a/tests/integration/partial/test_epi.py b/tests/integration/partial/test_epi.py index a5433b9..5922cf2 100644 --- a/tests/integration/partial/test_epi.py +++ b/tests/integration/partial/test_epi.py @@ -18,7 +18,7 @@ class TestEpi(BaseEpiTestSuite.EpiTestSuite): def set_expected_values(self): super().set_expected_values() - IntegrationTestUtils.TEST_START_DATE = '20221001' + IntegrationTestUtils.TEST_START_DATE = '20231127' IntegrationTestUtils.TEST_JOB_DIR = "EPI_" + IntegrationTestUtils.TEST_START_DATE diff --git a/tests/integration/partial/test_survey.py b/tests/integration/partial/test_survey.py index 0026f52..1c33aff 100644 --- a/tests/integration/partial/test_survey.py +++ b/tests/integration/partial/test_survey.py @@ -18,7 +18,7 @@ class TestSurvey(BaseSurveyTestSuite.SurveyTestSuite): IntegrationTestUtils.TEST_OUT_PATH = None def set_expected_values(self): - IntegrationTestUtils.TEST_START_DATE = '20221001' + IntegrationTestUtils.TEST_START_DATE = '20231127' IntegrationTestUtils.TEST_JOB_DIR = "SURVEYDATA_" + IntegrationTestUtils.TEST_START_DATE diff --git a/tests/integration/test_suites/depo_test_suite.py b/tests/integration/test_suites/depo_test_suite.py index 2b7c831..897462b 100644 --- a/tests/integration/test_suites/depo_test_suite.py +++ b/tests/integration/test_suites/depo_test_suite.py @@ -61,10 +61,10 @@ class BaseDepoTestSuite: def test_standard_run_all_input_csvs_produced(self): east_africa_csv_path = os.path.join(IntegrationTestUtils.TEST_OUT_PATH, IntegrationTestUtils.TEST_JOB_DIR, - "plotting", "eastafrica", "input_csvs", "*.tif") + "plotting", "eastafrica", "input_csvs", "*.nc") ethiopia_csv_path = os.path.join(IntegrationTestUtils.TEST_OUT_PATH, IntegrationTestUtils.TEST_JOB_DIR, - "plotting", "ethiopia", "input_csvs", "*.tif") + "plotting", "ethiopia", "input_csvs", "*.nc") ea_csv_count: int = IntegrationTestUtils.count_files_in_wildcard(east_africa_csv_path) eth_csv_count: int = IntegrationTestUtils.count_files_in_wildcard(ethiopia_csv_path) diff --git a/tests/integration/test_suites/env_suit_test_suite.py b/tests/integration/test_suites/env_suit_test_suite.py index 0030bfc..95cdca5 100644 --- a/tests/integration/test_suites/env_suit_test_suite.py +++ b/tests/integration/test_suites/env_suit_test_suite.py @@ -62,10 +62,10 @@ class BaseEnvSuitTestSuite: def test_standard_run_all_input_csvs_produced(self): east_africa_csv_path = os.path.join(IntegrationTestUtils.TEST_OUT_PATH, IntegrationTestUtils.TEST_JOB_DIR, - "plotting", "eastafrica", "input_csvs", "*.tif") + "plotting", "eastafrica", "input_csvs", "*.nc") ethiopia_csv_path = os.path.join(IntegrationTestUtils.TEST_OUT_PATH, IntegrationTestUtils.TEST_JOB_DIR, - "plotting", "ethiopia", "input_csvs", "*.tif") + "plotting", "ethiopia", "input_csvs", "*.nc") ea_csv_count: int = IntegrationTestUtils.count_files_in_wildcard(east_africa_csv_path) eth_csv_count: int = IntegrationTestUtils.count_files_in_wildcard(ethiopia_csv_path) -- GitLab