"""Injection well unit: injectivity and bottomhole pressure modules."""
from gemini_framework.abstract.unit_abstract import UnitAbstract
from gemini_framework.modules.injectionwell.calculate_bottomhole_pressure import (
CalculateBottomholePressure,
)
from gemini_framework.modules.injectionwell.calculate_injectivity_index import (
CalculateInjectivityIndex,
)
[docs]
class InjectionWellUnit(UnitAbstract):
"""Injection well unit includes bottomhole pressure and injectivity modules."""
def __init__(self, unit_id, unit_name, plant):
"""Initialize injection well unit."""
super().__init__(unit_id=unit_id, unit_name=unit_name, plant=plant)
# define unit modules
self.modules["model"].append(CalculateBottomholePressure(self))
self.modules["model"].append(CalculateInjectivityIndex(self))