diff --git a/emission/tests/analysisTests/intakeTests/TestPipelineCornerCases.py b/emission/tests/analysisTests/intakeTests/TestPipelineCornerCases.py index 56075b83b..4163949a7 100644 --- a/emission/tests/analysisTests/intakeTests/TestPipelineCornerCases.py +++ b/emission/tests/analysisTests/intakeTests/TestPipelineCornerCases.py @@ -49,7 +49,11 @@ def testNoData(self): ewps.PipelineStages.ALTERNATIVES, ewps.PipelineStages.USER_MODEL, ewps.PipelineStages.RECOMMENDATION, - ewps.PipelineStages.OUTPUT_GEN] + ewps.PipelineStages.OUTPUT_GEN, + ewps.PipelineStages.EXPORT_DATA, + ewps.PipelineStages.PURGE_TIMESERIES_DATA, + ewps.PipelineStages.RESTORE_TIMESERIES_DATA + ] for pse in ewps.PipelineStages.__iter__(): if pse in stages_skipped_in_testing: @@ -80,7 +84,11 @@ def testSkipPipelineNoNewEntries(self): ewps.PipelineStages.ALTERNATIVES, ewps.PipelineStages.USER_MODEL, ewps.PipelineStages.RECOMMENDATION, - ewps.PipelineStages.OUTPUT_GEN] + ewps.PipelineStages.OUTPUT_GEN, + ewps.PipelineStages.EXPORT_DATA, + ewps.PipelineStages.PURGE_TIMESERIES_DATA, + ewps.PipelineStages.RESTORE_TIMESERIES_DATA + ] test_run_states = list([pse.value for pse in filter(lambda pse: pse not in stages_skipped_in_testing, ewps.PipelineStages.__iter__())])