friction_loss.utils.plugin_dataclasses

 1from dataclasses import dataclass
 2from typing import Optional, List, Union
 3
 4import numpy as np
 5
 6PRECISION = 1e-5
 7
 8
 9@dataclass
10class PolypresInputParameter:
11    name: str  # Parameter name
12    value: Union[float, np.ndarray, str, None]  # Parameter value
13    unit: Optional[str]  # Parameter unit
14
15
16@dataclass
17class TransientData:
18    pressure: Union[List, np.ndarray]  # Pressure points array
19    qo: Union[float, np.ndarray]  # Value oil flow rate
20    qg: Union[float, np.ndarray]  # Value gas flow rate
21    qw: Union[float, np.ndarray]  # Value water flow rate
22    pressure_unit: str  # Pressure unit
23    qo_unit: str  # Unit oil flow rate
24    qg_unit: str  # Unit gas flow rate
25    qw_unit: str  # Unit water flow rate
26
27
28@dataclass
29class ProjectInfo:
30    analyst: str
31    analyst_email: str
32    company: str
33    contractor: str
34    country: str
35    curator: str
36    curator_email: str
37    field: str
38    id: str
39    info_date: str
40    project_name: str
41    well_name: str
42    test_name: str
43
44
45@dataclass
46class Pump:
47    pumpsize: float
48    f_nominal: float
49    f_actual: float
50    q_nominal: float
51    dp_nominal: float
52    k_f: float
53    stages_count: float
54    custom_stages_count: int
55    pump_koeffs: List[float]
56    hp_koeffs: List[float]
57
58    def equality_check(self, other_pump) -> bool:
59        pump_values = np.array([i for i in vars(self).values() if not isinstance(i, list)][:-1])
60        other_pump_values = np.array([i for i in vars(other_pump).values() if not isinstance(i, list)][:-1])
61
62        if (np.allclose(pump_values, other_pump_values, PRECISION) and
63                np.allclose(np.array(self.pump_koeffs), np.array(other_pump.pump_koeffs), PRECISION) and
64                np.allclose(np.array(self.hp_koeffs), np.array(other_pump.hp_koeffs), PRECISION)):
65            return True
66        else:
67            return False