Add a test to ensure that only the required data gets linked

Author: xylar

mpas_analysis/test/test_mpas_climatology_task.py

@@ -23,7 +23,7 @@
 
 from mpas_analysis.test import TestCase, loaddatadir
 from mpas_analysis.shared.climatology import MpasClimatologyTask, \
-    RemapMpasClimatologySubtask
+    RefYearMpasClimatologyTask, RemapMpasClimatologySubtask
 from mpas_analysis.shared import AnalysisTask
 from mpas_analysis.shared.analysis_task import \
     update_time_bounds_from_file_names
@@ -171,6 +171,36 @@ def test_update_climatology_bounds_and_create_symlinks(self):
             update_time_bounds_from_file_names(config, 'climatology', 'ocean',
                                                allow_cache=False)
 
+    def test_create_symlinks_isolates_reference_year_files(self):
+        mpasClimatologyTask = self.setup_task()
+
+        refYearTask = RefYearMpasClimatologyTask(
+            config=mpasClimatologyTask.config, componentName='ocean')
+        refYearTask.historyStreams = mpasClimatologyTask.historyStreams
+        refYearTask.startYear = 1
+        refYearTask.endYear = 1
+        refYearTask.inputFiles = []
+
+        for month in range(1, 13):
+            fileName = os.path.join(
+                self.test_dir,
+                f'mpaso.hist.am.timeSeriesStatsMonthly.0001-{month:02d}-01.nc')
+            with open(fileName, 'w'):
+                pass
+            refYearTask.inputFiles.append(fileName)
+
+        refSymlinkDirectory = refYearTask._create_symlinks()
+        mainSymlinkDirectory = mpasClimatologyTask._create_symlinks()
+
+        assert(refSymlinkDirectory != mainSymlinkDirectory)
+
+        mainSymlinkFiles = sorted(os.listdir(mainSymlinkDirectory))
+
+        assert(len(mainSymlinkFiles) == 12)
+        for fileName in mainSymlinkFiles:
+            assert(fileName.startswith(
+                'mpaso.hist.am.timeSeriesStatsMonthly.0002-'))
+
     def test_subtask_run_analysis(self):
         mpasClimatologyTask = self.setup_task()
         self.add_variables(mpasClimatologyTask)