diff --git a/tests/test_parsing.py b/tests/test_parsing.py index f36b76d..fd7a80b 100644 --- a/tests/test_parsing.py +++ b/tests/test_parsing.py @@ -16,17 +16,15 @@ from antares.tsgen.cluster_parsing import parse_cluster_ts, parse_yaml_cluster from antares.tsgen.cluster_resolve import resolve_thermal_cluster -from antares.tsgen.ts_generator import ProbabilityLaw, ThermalCluster +from antares.tsgen.ts_generator import OutageGenerationParameters, ProbabilityLaw, ThermalCluster NB_OF_DAY = 10 @pytest.fixture def cluster() -> ThermalCluster: - return ThermalCluster( + outage_gen_params = OutageGenerationParameters( unit_count=1, - nominal_power=500, - modulation=np.ones(dtype=float, shape=8760), fo_law=ProbabilityLaw.UNIFORM, fo_volatility=0, po_law=ProbabilityLaw.UNIFORM, @@ -38,6 +36,11 @@ def cluster() -> ThermalCluster: npo_min=np.zeros(dtype=int, shape=NB_OF_DAY), npo_max=np.ones(dtype=int, shape=NB_OF_DAY), ) + return ThermalCluster( + outage_gen_params, + nominal_power=500, + modulation=np.ones(dtype=float, shape=8760), + ) def test(cluster, data_directory):