def test_get_lead_sequence_init(metplus_config, current_hour, lead_seq):
input_dict = {'valid': datetime.datetime(2019, 2, 1, current_hour)}
conf = metplus_config()
conf.set('config', 'INIT_SEQ', "0, 12")
conf.set('config', 'LEAD_SEQ_MAX', 36)
test_seq = util.get_lead_sequence(conf, input_dict)
assert (test_seq == [relativedelta(hours=lead) for lead in lead_seq])
def test_get_lead_sequence_init_min_10(metplus_config):
input_dict = {'valid': datetime.datetime(2019, 2, 1, 12)}
conf = metplus_config()
conf.set('config', 'INIT_SEQ', "0, 12")
conf.set('config', 'LEAD_SEQ_MAX', 24)
conf.set('config', 'LEAD_SEQ_MIN', 10)
test_seq = util.get_lead_sequence(conf, input_dict)
lead_seq = [12, 24]
assert (test_seq == [relativedelta(hours=lead) for lead in lead_seq])
def test_get_lead_sequence_lead(metplus_config):
input_dict = {'valid': datetime.datetime(2019, 2, 1, 13)}
conf = metplus_config()
conf.set('config', 'LEAD_SEQ', "3,6,9,12")
test_seq = util.get_lead_sequence(conf, input_dict)
hour_seq = []
for test in test_seq:
hour_seq.append(time_util.ti_get_hours_from_relativedelta(test))
lead_seq = [3, 6, 9, 12]
assert (hour_seq == lead_seq)
def test_get_lead_sequence_groups(metplus_config, config_dict, expected_list):
config = metplus_config()
for key, value in config_dict.items():
config.set('config', key, value)
output_list = util.get_lead_sequence(config)
hour_seq = []
for output in output_list:
hour_seq.append(time_util.ti_get_hours_from_relativedelta(output))
assert (hour_seq == expected_list)
def test_get_lead_sequence_lead_list(key, value):
input_dict = {'valid': datetime.datetime(2019, 2, 1, 13)}
conf = metplus_config()
conf.set('config', 'LEAD_SEQ', key)
test_seq = util.get_lead_sequence(conf, input_dict)
hour_seq = []
for test in test_seq:
hour_seq.append(
time_util.ti_get_seconds_from_relativedelta(test) // 3600)
lead_seq = value
assert (hour_seq == lead_seq)