Coverage for rivapy/models/__init__.py: 96%

1 

2import numpy as np 

3from numpy.core.fromnumeric import var 

4from rivapy.models.factory import _factory 

5from rivapy.models.local_vol import LocalVol 

6from rivapy.models.heston import HestonModel 

7from rivapy.models.stoch_local_vol import StochasticLocalVol 

8from rivapy.models.scott_chesney import ScottChesneyModel 

9from rivapy.models.ornstein_uhlenbeck import OrnsteinUhlenbeck 

10from rivapy.models.lucia_schwartz import LuciaSchwartz 

11from rivapy.models.residual_demand_model import ResidualDemandModel, WindPowerModel, SolarPowerModel, SupplyFunction, LoadModel, SmoothstepSupplyCurve 

12from rivapy.models.residual_demand_fwd_model import WindPowerForecastModel, WindPowerForecastModelParameter, ResidualDemandForwardModel, MultiRegionWindForecastModel, LinearDemandForwardModel 

13 

14def _add_to_factory(cls): 

15 factory_entries = _factory() 

16 factory_entries[cls.__name__] = cls 

17 

18_add_to_factory(OrnsteinUhlenbeck) 

19_add_to_factory(LuciaSchwartz) 

20_add_to_factory(SupplyFunction) 

21_add_to_factory(SmoothstepSupplyCurve) 

22_add_to_factory(WindPowerForecastModel) 

23_add_to_factory(WindPowerForecastModelParameter) 

24_add_to_factory(ResidualDemandForwardModel) 

25_add_to_factory(MultiRegionWindForecastModel.Region) 

26_add_to_factory(MultiRegionWindForecastModel) 

27 

28 

29 

30if __name__=='__main__': 

31 pass