Add convenient properties: nInj, nProd

patnr · Sep 22, 2023 · 34323da · 34323da
1 parent d02f116
commit 34323da
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/TPFA_ResSim/__init__.py b/TPFA_ResSim/__init__.py
@@ -86,6 +86,11 @@ def __setattr__(self, key, val):
     por: np.ndarray = None
     """Porosity; Array of shape `(Nx, Ny)`)."""
 
+    nInj  = property(lambda self: len(self.inj_xy))
+    """Num. of injector wells."""
+    nProd = property(lambda self: len(self.prod_xy))
+    """Num. of producer wells."""
+
     inj_xy: np.ndarray = None
     """Array of shape `(nWell, 2)` of x- and y-coords for `nWell` injectors.