def test_create_nwp_forecasts_outside(template_fx, mocker, log): vars_ = ('ac_power', 'dni') api, template, site = template_fx site = site.replace(latitude=19, longitude=-159) templates = [template.replace(name='one'), template.replace(name='two')] with pytest.raises(ValueError): common.create_nwp_forecasts(api, site, vars_, templates)
def test_create_nwp_forecasts(template_fx, mocker, vars_, primary): api, template, site = template_fx templates = [template.replace(name='one'), template.replace(name='two')] fxdict = template.to_dict() fxdict['constant_values'] = [0, 50, 100] fxdict['axis'] = 'y' templates += [ProbabilisticForecast.from_dict(fxdict)] fxs = common.create_nwp_forecasts(api, site, vars_, templates) assert len(fxs) == 6 if primary: assert fxs[0].variable == primary assert fxs[2].variable == primary assert 'one' in fxs[0].name assert 'one' in fxs[1].name assert fxs[0].forecast_id in fxs[1].extra_parameters assert 'two' in fxs[2].name assert 'two' in fxs[3].name assert fxs[2].forecast_id in fxs[3].extra_parameters assert isinstance(fxs[-1], ProbabilisticForecast)