Coverage for backend/idaes_factory/adapters/adapter_library.py: 100%

16 statements  

« prev     ^ index     » next       coverage.py v7.10.7, created at 2025-11-06 23:27 +0000

1from .core_adapters import * 

2from .generic_adapters import * 

3from .dynamic_adapter import * 

4from common.config_types import * 

5from .generic_adapters import * 

6from .dynamic_adapter import * 

7from common.config_types import * 

8from flowsheetInternals.unitops.models import * 

9from .unit_models import * 

10from .unit_models.ml_adapter import * 

11from .toggle_adapter import * 

12from .enum_adapter import * 

13from .unit_models.ml_adapter import * 

14from .toggle_adapter import * 

15from .enum_adapter import * 

16 

17adapters: dict[str, UnitModelAdapter] = { 

18 "pump": pump_adapter, 

19 "compressor": compressor_adapter, 

20 "turbine": turbine_adapter, 

21 "willans_turbine": willans_turbine_adapter, 

22 "valve": valve_adapter, 

23 "heater": heater_adapter, 

24 "Tank": tank_adapter, 

25 "mixer": mixer_adapter, 

26 "cooler": cooler_adapter, 

27 # "recycle": recycle_adapter, 

28 "splitter": splitter_adapter, 

29 "header": header_adapter, 

30 "phaseSeparator": phase_separator_adapter, 

31 "compoundSeparator": compound_separator_adapter, 

32 "heatExchanger": heat_exchanger_adapter, 

33 "heatPump": heat_pump_adapter, 

34 "machineLearningBlock": ml_adapter, 

35 "link": link_adapter, 

36 "solar": solar_adapter, 

37 "wind" : wind_adapter, 

38 "energy_mixer": energy_mixer_adapter, 

39 "direct_steam_injection": direct_steam_injection_adapter, 

40 "bus": bus_adapter, 

41 "acBus": ac_bus_adapter, 

42 "transmissionLine": transmission_line_adapter, 

43 "load": load_adapter, 

44 "transformer": transformer_adapter, 

45 "storage": storage_adapter, 

46 "translator": translator_adapter, 

47 "hydro": hydro_adapter, 

48 "pid_controller": pid_adapter, 

49 "custom_variable": pid_adapter, # Just like pid, the custom variable has no adapter 

50 "grid": grid_adapter, 

51 "integration": integration_adapter, 

52 "RCT_CSTR": reactor_adapter, 

53 "energy_splitter": energy_splitter_adapter, 

54 "mdb": mdb_adapter, 

55 "pl_turbine": pl_willans_turbine_adapter, 

56 "dts_turbine": tsat_willans_turbine_adapter, 

57 "bs_turbine": bs_willans_turbine_adapter, 

58 "cs_turbine": cs_willans_turbine_adapter, 

59 "simple_header": simple_header_adapter, 

60 "stoich_hda": reactor_adapter, # doesn't need anything special 

61 "heat_exchanger_1d": heat_exchanger_1d_adapter, 

62 "steam_user" : steam_user_adapter, 

63 "desuperheater": desuperheater_adapter, 

64 "heat_exchanger_ntu": ntu_heat_exchanger_adapter, 

65 "waterpipe": waterpipe_adapter, 

66 "heat_exchanger_lc": lumped_capacitance_heat_exchanger_adapter 

67} 

68 

69 

70# ArgAdapterTypes: dict= { 

71# (ArgType.PropertyPackage, PropertyPackageAdapter()), 

72# (ArgType.SteamPropertyPackage, PropertyPackageAdapter()), 

73# (ArgType.InletPropertyPackage, PropertyPackageAdapter()), 

74# (ArgType.OutletPropertyPackage, PropertyPackageAdapter()), 

75# (ArgType.NumInlets, NumInletsAdapter()), 

76# (ArgType.NumOutlets, NumOutletsAdapter()), 

77# (ArgType.HotSide, ArgAdapter("Hot Side")), 

78# (ArgType.ColdSide, ArgAdapter("Cold Side")), 

79# (ArgType.Dynamic, DynamicAdapter()), 

80# (ArgType.HasPressureChange, ConstantArg(True)), 

81# (ArgType.HasHoldUp, DynamicAdapter()), 

82# (ArgType.Model, JSONModelAdapter()), 

83# (ArgType.IDs, IDAdapter()), 

84# (ArgType.UnitOpNames, unitopNamesAdapter()), 

85# (ArgType.EnableCoefficients, ToggleAdapter("enable_coefficients")), 

86# (ArgType.ValveFunction, EnumAdapter("valve_function")), 

87# }