def test_run_period():
"""Test the run_period command."""
runner = CliRunner()
result = runner.invoke(run_period, ['1', '6', '1', '12'])
assert result.exit_code == 0
run_per = RunPeriod.from_string(result.output)
assert run_per.start_date == Date(1, 6)
assert run_per.end_date == Date(1, 12)
def test_run_period_string_methods():
"""Test the to/from string methods."""
run_period = RunPeriod()
run_period.start_date = Date(1, 1)
run_period.end_date = Date(6, 21)
run_period.start_day_of_week = 'Monday'
run_period.holidays = (Date(1, 1), Date(3, 17))
run_period.daylight_saving_time = DaylightSavingTime()
new_run_period = RunPeriod.from_string(str(run_period))
assert new_run_period == run_period
def _load_run_period_str(run_period_str):
"""Load a RunPeriod from a string of a run period or analysis period.
Args:
run_period_str: A string of a RunPeriod or AnalysisPeriod to be loaded.
"""
if run_period_str is not None and run_period_str != '' \
and run_period_str != 'None':
if run_period_str.startswith('RunPeriod'):
return RunPeriod.from_string(run_period_str)
else:
return RunPeriod.from_analysis_period(
AnalysisPeriod.from_string(run_period_str))
