add all tests

AntaresSimulatorTeam · Mar 6, 2025 · 26d1bc2 · 26d1bc2
1 parent 9e8bea6
commit 26d1bc2
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 63 deletions.
diff --git a/tests/antares/services/local_services/test_constraints.py b/tests/antares/services/local_services/test_constraints.py
@@ -81,15 +81,12 @@ def test_matrices(self, local_study_w_constraints: Study) -> None:
 
         # Replace matrices
         matrix = pd.DataFrame(data=8784 * [[3]])
-        assert bc.get_greater_term_matrix().empty
         bc.update_greater_term_matrix(matrix)
         assert bc.get_greater_term_matrix().equals(matrix)
 
-        assert bc.get_less_term_matrix().empty
         bc.update_less_term_matrix(matrix)
         assert bc.get_less_term_matrix().equals(matrix)
 
-        assert bc.get_equal_term_matrix().empty
         bc.update_equal_term_matrix(matrix)
         assert bc.get_equal_term_matrix().equals(matrix)
 

diff --git a/tests/antares/services/local_services/test_default_matrices.py b/tests/antares/services/local_services/test_default_matrices.py
@@ -9,8 +9,10 @@
 # SPDX-License-Identifier: MPL-2.0
 #
 # This file is part of the Antares project.
+import numpy as np
 import pandas as pd
 
+from antares.craft.model.binding_constraint import BindingConstraintFrequency, BindingConstraintPropertiesUpdate
 from antares.craft.tools.matrix_tool import (
     default_credit_modulation,
     default_data_matrix,
@@ -33,78 +35,55 @@ class TestDefaultMatrices:
     def test_all_matrices(self, local_study_with_constraint):
         area_fr = local_study_with_constraint.get_areas()["fr"]
         # Load
-        load = area_fr.get_load_matrix()
-        assert load.equals(pd.DataFrame(default_series))
+        assert area_fr.get_load_matrix().equals(pd.DataFrame(default_series))
         # Misc-gen
-        misc_gen = area_fr.get_misc_gen_matrix()
-        assert misc_gen.equals(pd.DataFrame(default_mingen))
+        assert area_fr.get_misc_gen_matrix().equals(pd.DataFrame(default_mingen))
         # Reserves
-        reserves = area_fr.get_reserves_matrix()
-        assert reserves.equals(pd.DataFrame(default_reserves))
+        assert area_fr.get_reserves_matrix().equals(pd.DataFrame(default_reserves))
         # Wind
-        wind = area_fr.get_wind_matrix()
-        assert wind.equals(pd.DataFrame(default_series))
+        assert area_fr.get_wind_matrix().equals(pd.DataFrame(default_series))
         # Solar
-        solar = area_fr.get_solar_matrix()
-        assert solar.equals(pd.DataFrame(default_series))
+        assert area_fr.get_solar_matrix().equals(pd.DataFrame(default_series))
         # Hydro
-        mod = area_fr.hydro.get_mod_series()
-        assert mod.equals(pd.DataFrame(default_mod))
-        ror = area_fr.hydro.get_ror_series()
-        assert ror.equals(pd.DataFrame(default_series))
-        energy = area_fr.hydro.get_energy()
-        assert energy.equals(pd.DataFrame(default_energy))
-        hydro_mingen = area_fr.hydro.get_mingen()
-        assert hydro_mingen.equals(pd.DataFrame(default_series))
-        reservoir = area_fr.hydro.get_reservoir()
-        assert reservoir.equals(pd.DataFrame(default_reservoir))
-        credit_modulation = area_fr.hydro.get_credit_modulations()
-        assert credit_modulation.equals(pd.DataFrame(default_credit_modulation))
-        hydro_max_power = area_fr.hydro.get_maxpower()
-        assert hydro_max_power.equals(pd.DataFrame(default_maxpower))
-        hydro_inflow = area_fr.hydro.get_inflow_pattern()
-        assert hydro_inflow.equals(pd.DataFrame(default_inflow_pattern))
-        water_values = area_fr.hydro.get_water_values()
-        assert water_values.equals(pd.DataFrame(default_water_values))
+        assert area_fr.hydro.get_mod_series().equals(pd.DataFrame(default_mod))
+        assert area_fr.hydro.get_ror_series().equals(pd.DataFrame(default_series))
+        assert area_fr.hydro.get_energy().equals(pd.DataFrame(default_energy))
+        assert area_fr.hydro.get_mingen().equals(pd.DataFrame(default_series))
+        assert area_fr.hydro.get_reservoir().equals(pd.DataFrame(default_reservoir))
+        assert area_fr.hydro.get_credit_modulations().equals(pd.DataFrame(default_credit_modulation))
+        assert area_fr.hydro.get_maxpower().equals(pd.DataFrame(default_maxpower))
+        assert area_fr.hydro.get_inflow_pattern().equals(pd.DataFrame(default_inflow_pattern))
+        assert area_fr.hydro.get_water_values().equals(pd.DataFrame(default_water_values))
         # Binding constraint
-        # todo
         bc = local_study_with_constraint.get_binding_constraints()["test constraint"]
-        bc.get_equal_term_matrix()
-        bc.get_greater_term_matrix()
-        bc.get_less_term_matrix()
+        default_hourly_matrix = pd.DataFrame(np.zeros((8784, 1)))
+        assert bc.get_equal_term_matrix().equals(default_hourly_matrix)
+        assert bc.get_greater_term_matrix().equals(default_hourly_matrix)
+        assert bc.get_less_term_matrix().equals(default_hourly_matrix)
+        bc.update_properties(BindingConstraintPropertiesUpdate(time_step=BindingConstraintFrequency.DAILY))
+        default_daily_matrix = pd.DataFrame(np.zeros((366, 1)))
+        assert bc.get_equal_term_matrix().equals(default_daily_matrix)
+        assert bc.get_greater_term_matrix().equals(default_daily_matrix)
+        assert bc.get_less_term_matrix().equals(default_daily_matrix)
         # Links
         link = local_study_with_constraint.get_links()["at / fr"]
-        parameters = link.get_parameters()
-        assert parameters.equals(pd.DataFrame(default_link_parameters))
-        capa_direct = link.get_capacity_direct()
-        assert capa_direct.equals(pd.DataFrame(default_series_with_ones))
-        capa_indirect = link.get_capacity_indirect()
-        assert capa_indirect.equals(pd.DataFrame(default_series_with_ones))
+        assert link.get_parameters().equals(pd.DataFrame(default_link_parameters))
+        assert link.get_capacity_direct().equals(pd.DataFrame(default_series_with_ones))
+        assert link.get_capacity_indirect().equals(pd.DataFrame(default_series_with_ones))
         # Renewables
         renewable = area_fr.get_renewables()["renewable cluster"]
-        renewable_series = renewable.get_timeseries()
-        assert renewable_series.equals(pd.DataFrame(default_series))
+        assert renewable.get_timeseries().equals(pd.DataFrame(default_series))
         # St storages
         storage = area_fr.get_st_storages()["short term storage"]
-        storage_inflow = storage.get_storage_inflows()
-        assert storage_inflow.equals(pd.DataFrame(default_series))
-        pmax_injection = storage.get_pmax_injection()
-        assert pmax_injection.equals(pd.DataFrame(default_series_with_ones))
-        pmax_withdrawal = storage.get_pmax_withdrawal()
-        assert pmax_withdrawal.equals(pd.DataFrame(default_series_with_ones))
-        lower_curve = storage.get_lower_rule_curve()
-        assert lower_curve.equals(pd.DataFrame(default_series))
-        upper_curve = storage.get_upper_rule_curve()
-        assert upper_curve.equals(pd.DataFrame(default_series_with_ones))
+        assert storage.get_storage_inflows().equals(pd.DataFrame(default_series))
+        assert storage.get_pmax_injection().equals(pd.DataFrame(default_series_with_ones))
+        assert storage.get_pmax_withdrawal().equals(pd.DataFrame(default_series_with_ones))
+        assert storage.get_lower_rule_curve().equals(pd.DataFrame(default_series))
+        assert storage.get_upper_rule_curve().equals(pd.DataFrame(default_series_with_ones))
         # Thermal
         thermal = area_fr.get_thermals()["test thermal cluster"]
-        thermal_series = thermal.get_series_matrix()
-        assert thermal_series.equals(pd.DataFrame(default_series))
-        prepro_data = thermal.get_prepro_data_matrix()
-        assert prepro_data.equals(pd.DataFrame(default_data_matrix))
-        prepro_modulation = thermal.get_prepro_modulation_matrix()
-        assert prepro_modulation.equals(pd.DataFrame(default_modulation_matrix))
-        fuel_matrix = thermal.get_fuel_cost_matrix()
-        assert fuel_matrix.equals(pd.DataFrame(default_series))
-        co2_matrix = thermal.get_co2_cost_matrix()
-        assert co2_matrix.equals(pd.DataFrame(default_series))
+        assert thermal.get_series_matrix().equals(pd.DataFrame(default_series))
+        assert thermal.get_prepro_data_matrix().equals(pd.DataFrame(default_data_matrix))
+        assert thermal.get_prepro_modulation_matrix().equals(pd.DataFrame(default_modulation_matrix))
+        assert thermal.get_fuel_cost_matrix().equals(pd.DataFrame(default_series))
+        assert thermal.get_co2_cost_matrix().equals(pd.DataFrame(default_series))