1818 calculate_polytropic_head_campbell ,
1919)
2020from libecalc .domain .process .compressor .dto import CompressorStage
21+ from libecalc .domain .process .value_objects .chart .compressor .compressor_chart_dto import CompressorChart
2122from libecalc .domain .process .value_objects .chart .generic import GenericChartFromDesignPoint , GenericChartFromInput
2223from libecalc .domain .process .value_objects .fluid_stream .fluid_model import FluidModel
2324from libecalc .infrastructure .neqsim_fluid_provider .neqsim_fluid_factory import NeqSimFluidFactory
24- from libecalc .domain .process .value_objects .chart .compressor .compressor_chart_dto import CompressorChart
2525
2626
2727@pytest .fixture
@@ -192,7 +192,7 @@ def test_simplified_compressor_train_unknown_stages(
192192 fluid_model = fluid_model_rich , chart = variable_speed_compressor_chart_dto
193193 )
194194 # Prepare stages using centralized builder method (correct data flow)
195- SimplifiedTrainBuilder .prepare_model_stages_from_data (
195+ SimplifiedTrainBuilder .prepare_compressor_stages_from_data (
196196 compressor_model = compressor_train ,
197197 rate = np .linspace (start = 1000 , stop = 10000 , num = 10 ),
198198 suction_pressure = np .linspace (start = 10 , stop = 20 , num = 10 ),
@@ -213,7 +213,7 @@ def test_simplified_compressor_train_unknown_stages_with_constant_power_adjustme
213213 fluid_model = fluid_model_rich , chart = variable_speed_compressor_chart_dto
214214 )
215215 # Prepare stages using centralized builder method (correct data flow)
216- SimplifiedTrainBuilder .prepare_model_stages_from_data (
216+ SimplifiedTrainBuilder .prepare_compressor_stages_from_data (
217217 compressor_model = compressor_train_energy_function ,
218218 rate = np .linspace (start = 1000 , stop = 10000 , num = 10 ),
219219 suction_pressure = np .linspace (start = 10 , stop = 20 , num = 10 ),
@@ -400,7 +400,7 @@ def test_compressor_train_simplified_known_stages_generic_chart(
400400
401401 # Make the undefined compressor chart, using rate and pressure input
402402 # Prepare stages using centralized builder method (correct data flow)
403- SimplifiedTrainBuilder .prepare_model_stages_from_data (
403+ SimplifiedTrainBuilder .prepare_compressor_stages_from_data (
404404 compressor_model = simple_compressor_train_model_extra_generic_stage_from_data ,
405405 rate = rates ,
406406 suction_pressure = suction_pressures ,
@@ -476,7 +476,7 @@ def test_compressor_train_simplified_unknown_stages(
476476 )
477477
478478 # Prepare stages using centralized builder method (correct data flow)
479- SimplifiedTrainBuilder .prepare_model_stages_from_data (
479+ SimplifiedTrainBuilder .prepare_compressor_stages_from_data (
480480 compressor_model = simple_compressor_train_model ,
481481 rate = rates ,
482482 suction_pressure = suction_pressures ,
@@ -858,7 +858,7 @@ def test_calculate_compressor_work(fluid_factory_medium, simplified_compressor_t
858858 stages = stages , fluid_model = fluid_factory2 .fluid_model
859859 )
860860 # Prepare stages using centralized builder method (correct data flow)
861- SimplifiedTrainBuilder .prepare_model_stages_from_data (
861+ SimplifiedTrainBuilder .prepare_compressor_stages_from_data (
862862 compressor_model = compressor_train ,
863863 rate = fluid_factory2 .mass_rate_to_standard_rate (mass_rates ),
864864 suction_pressure = inlet_pressures ,
0 commit comments