open-fdd 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

open_fdd/air_handling_unit/faults/fault_condition.py

@@ -1,21 +1,39 @@
+import pandas as pd
 import pandas.api.types as pdtypes
 from open_fdd.air_handling_unit.faults.helper_utils import HelperUtils
 import sys
 
 
+class MissingColumnError(Exception):
+    """Custom exception raised when a required column is missing or None."""
+
+    def __init__(self, message):
+        self.message = message
+        super().__init__(self.message)
+
+
 class FaultCondition:
     """Parent class for Fault Conditions. Methods are inherited to all children."""
 
-    def set_attributes(self, dict_):
-        """Passes dictionary into initialization of class instance, then uses the attributes called out below in
-        attributes_dict to set only the attributes that match from dict_.
+    def __init__(self):
+        self.required_columns = []
 
-        :param dict_: dictionary of all possible class attributes (loaded from config file)
-        """
-        for attribute in self.__dict__:
+    def set_attributes(self, dict_):
+        """Passes dictionary into initialization of class instance"""
+        for attribute in self.__dict__.keys():
             upper = attribute.upper()
-            value = dict_[upper]
-            self.__setattr__(attribute, value)
+            if upper in dict_:
+                value = dict_[upper]
+                self.__setattr__(attribute, value)
+
+    def check_required_columns(self, df: pd.DataFrame):
+        """Checks if required columns are present in the DataFrame."""
+        missing_columns = [
+            col for col in self.required_columns if col is None or col not in df.columns
+        ]
+
+        if missing_columns:
+            raise MissingColumnError(f"Missing required columns: {missing_columns}")
 
     def troubleshoot_cols(self, df):
         """print troubleshoot columns mapping

open_fdd/air_handling_unit/faults/fault_condition_eight.py

@@ -1,7 +1,9 @@
 import pandas as pd
 import numpy as np
-from open_fdd.air_handling_unit.faults.fault_condition import FaultCondition
-from open_fdd.air_handling_unit.faults.helper_utils import HelperUtils
+from open_fdd.air_handling_unit.faults.fault_condition import (
+    FaultCondition,
+    MissingColumnError,
+)
 import sys
 
 
@@ -12,6 +14,7 @@ class FaultConditionEight(FaultCondition):
     """
 
     def __init__(self, dict_):
+        super().__init__()
         self.delta_t_supply_fan = float
         self.mix_degf_err_thres = float
         self.supply_degf_err_thres = float
@@ -25,43 +28,64 @@ class FaultConditionEight(FaultCondition):
 
         self.set_attributes(dict_)
 
-    def apply(self, df: pd.DataFrame) -> pd.DataFrame:
-        if self.troubleshoot_mode:
-            self.troubleshoot_cols(df)
-
-        # Check analog outputs [data with units of %] are floats only
-        columns_to_check = [
+        # Set required columns specific to this fault condition
+        self.required_columns = [
+            self.mat_col,
+            self.sat_col,
             self.economizer_sig_col,
             self.cooling_sig_col,
         ]
 
-        self.check_analog_pct(df, columns_to_check)
-
-        df["sat_fan_mat"] = abs(
-            df[self.sat_col] - self.delta_t_supply_fan - df[self.mat_col]
-        )
-        df["sat_mat_sqrted"] = np.sqrt(
-            self.supply_degf_err_thres**2 + self.mix_degf_err_thres**2
-        )
-
-        df["combined_check"] = (
-            (df["sat_fan_mat"] > df["sat_mat_sqrted"])
-            & (df[self.economizer_sig_col] > self.ahu_min_oa_dpr)
-            & (df[self.cooling_sig_col] < 0.1)
-        )
-
-        # Rolling sum to count consecutive trues
-        rolling_sum = (
-            df["combined_check"].rolling(window=self.rolling_window_size).sum()
-        )
-        # Set flag to 1 if rolling sum equals the window size
-        df["fc8_flag"] = (rolling_sum >= self.rolling_window_size).astype(int)
-
-        if self.troubleshoot_mode:
-            print("Troubleshoot mode enabled - not removing helper columns")
-            sys.stdout.flush()
-            del df["sat_fan_mat"]
-            del df["sat_mat_sqrted"]
-            del df["combined_check"]
+    def get_required_columns(self) -> str:
+        """Returns a string representation of the required columns."""
+        return f"Required columns for FaultConditionEight: {', '.join(self.required_columns)}"
+
+    def apply(self, df: pd.DataFrame) -> pd.DataFrame:
+        try:
+            # Ensure all required columns are present
+            self.check_required_columns(df)
+
+            if self.troubleshoot_mode:
+                self.troubleshoot_cols(df)
+
+            # Check analog outputs [data with units of %] are floats only
+            columns_to_check = [
+                self.economizer_sig_col,
+                self.cooling_sig_col,
+            ]
 
-        return df
+            self.check_analog_pct(df, columns_to_check)
+
+            df["sat_fan_mat"] = abs(
+                df[self.sat_col] - self.delta_t_supply_fan - df[self.mat_col]
+            )
+            df["sat_mat_sqrted"] = np.sqrt(
+                self.supply_degf_err_thres**2 + self.mix_degf_err_thres**2
+            )
+
+            df["combined_check"] = (
+                (df["sat_fan_mat"] > df["sat_mat_sqrted"])
+                & (df[self.economizer_sig_col] > self.ahu_min_oa_dpr)
+                & (df[self.cooling_sig_col] < 0.1)
+            )
+
+            # Rolling sum to count consecutive trues
+            rolling_sum = (
+                df["combined_check"].rolling(window=self.rolling_window_size).sum()
+            )
+            # Set flag to 1 if rolling sum equals the window size
+            df["fc8_flag"] = (rolling_sum >= self.rolling_window_size).astype(int)
+
+            if self.troubleshoot_mode:
+                print("Troubleshoot mode enabled - not removing helper columns")
+                sys.stdout.flush()
+                del df["sat_fan_mat"]
+                del df["sat_mat_sqrted"]
+                del df["combined_check"]
+
+            return df
+
+        except MissingColumnError as e:
+            print(f"Error: {e.message}")
+            sys.stdout.flush()
+            raise e  # Re-raise the exception so it can be caught by pytest

open_fdd/air_handling_unit/faults/fault_condition_eleven.py

@@ -1,7 +1,8 @@
 import pandas as pd
-import numpy as np
-from open_fdd.air_handling_unit.faults.fault_condition import FaultCondition
-from open_fdd.air_handling_unit.faults.helper_utils import HelperUtils
+from open_fdd.air_handling_unit.faults.fault_condition import (
+    FaultCondition,
+    MissingColumnError,
+)
 import sys
 
 
@@ -9,11 +10,11 @@ class FaultConditionEleven(FaultCondition):
     """Class provides the definitions for Fault Condition 11.
     Outside air temperature too low for 100% outdoor
     air cooling in economizer cooling mode.
-
     Economizer performance fault
     """
 
     def __init__(self, dict_):
+        super().__init__()
         self.delta_t_supply_fan = float
         self.outdoor_degf_err_thres = float
         self.supply_degf_err_thres = float
@@ -26,43 +27,64 @@ class FaultConditionEleven(FaultCondition):
 
         self.set_attributes(dict_)
 
-    def apply(self, df: pd.DataFrame) -> pd.DataFrame:
-        if self.troubleshoot_mode:
-            self.troubleshoot_cols(df)
-
-        # Check analog outputs [data with units of %] are floats only
-        columns_to_check = [
-            self.economizer_sig_col,
+        # Set required columns specific to this fault condition
+        self.required_columns = [
+            self.sat_setpoint_col,
+            self.oat_col,
             self.cooling_sig_col,
+            self.economizer_sig_col,
         ]
-        self.check_analog_pct(df, columns_to_check)
 
-        df["oat_plus_oaterror"] = df[self.oat_col] + self.outdoor_degf_err_thres
-        df["satsp_delta_saterr"] = (
-            df[self.sat_setpoint_col]
-            - self.delta_t_supply_fan
-            - self.supply_degf_err_thres
-        )
+    def get_required_columns(self) -> str:
+        """Returns a string representation of the required columns."""
+        return f"Required columns for FaultConditionEleven: {', '.join(self.required_columns)}"
 
-        df["combined_check"] = (
-            (df["oat_plus_oaterror"] < df["satsp_delta_saterr"])
-            # verify ahu is running in OS 3 clg mode in 100 OA
-            & (df[self.cooling_sig_col] > 0.01)
-            & (df[self.economizer_sig_col] > 0.9)
-        )
+    def apply(self, df: pd.DataFrame) -> pd.DataFrame:
+        try:
+            # Ensure all required columns are present
+            self.check_required_columns(df)
 
-        # Rolling sum to count consecutive trues
-        rolling_sum = (
-            df["combined_check"].rolling(window=self.rolling_window_size).sum()
-        )
-        # Set flag to 1 if rolling sum equals the window size
-        df["fc11_flag"] = (rolling_sum >= self.rolling_window_size).astype(int)
+            if self.troubleshoot_mode:
+                self.troubleshoot_cols(df)
 
-        if self.troubleshoot_mode:
-            print("Troubleshoot mode enabled - not removing helper columns")
-            sys.stdout.flush()
-            del df["oat_plus_oaterror"]
-            del df["satsp_delta_saterr"]
-            del df["combined_check"]
+            # Check analog outputs [data with units of %] are floats only
+            columns_to_check = [
+                self.economizer_sig_col,
+                self.cooling_sig_col,
+            ]
+            self.check_analog_pct(df, columns_to_check)
 
-        return df
+            df["oat_plus_oaterror"] = df[self.oat_col] + self.outdoor_degf_err_thres
+            df["satsp_delta_saterr"] = (
+                df[self.sat_setpoint_col]
+                - self.delta_t_supply_fan
+                - self.supply_degf_err_thres
+            )
+
+            df["combined_check"] = (
+                (df["oat_plus_oaterror"] < df["satsp_delta_saterr"])
+                # verify ahu is running in OS 3 clg mode in 100 OA
+                & (df[self.cooling_sig_col] > 0.01)
+                & (df[self.economizer_sig_col] > 0.9)
+            )
+
+            # Rolling sum to count consecutive trues
+            rolling_sum = (
+                df["combined_check"].rolling(window=self.rolling_window_size).sum()
+            )
+            # Set flag to 1 if rolling sum equals the window size
+            df["fc11_flag"] = (rolling_sum >= self.rolling_window_size).astype(int)
+
+            if self.troubleshoot_mode:
+                print("Troubleshoot mode enabled - not removing helper columns")
+                sys.stdout.flush()
+                del df["oat_plus_oaterror"]
+                del df["satsp_delta_saterr"]
+                del df["combined_check"]
+
+            return df
+
+        except MissingColumnError as e:
+            print(f"Error: {e.message}")
+            sys.stdout.flush()
+            raise e  # Re-raise the exception so it can be caught by pytest

open_fdd/air_handling_unit/faults/fault_condition_fifteen.py

@@ -1,17 +1,20 @@
 import pandas as pd
 import numpy as np
-from open_fdd.air_handling_unit.faults.fault_condition import FaultCondition
-from open_fdd.air_handling_unit.faults.helper_utils import HelperUtils
+from open_fdd.air_handling_unit.faults.fault_condition import (
+    FaultCondition,
+    MissingColumnError,
+)
 import sys
 
 
 class FaultConditionFifteen(FaultCondition):
     """Class provides the definitions for Fault Condition 15.
-    Temperature rise across inactive heating coi.
+    Temperature rise across inactive heating coil.
     Requires coil leaving temp sensor.
     """
 
     def __init__(self, dict_):
+        super().__init__()
         self.delta_supply_fan = float
         self.coil_temp_enter_err_thres = float
         self.coil_temp_leav_err_thres = float
@@ -27,64 +30,87 @@ class FaultConditionFifteen(FaultCondition):
 
         self.set_attributes(dict_)
 
-    def apply(self, df: pd.DataFrame) -> pd.DataFrame:
-        if self.troubleshoot_mode:
-            self.troubleshoot_cols(df)
-
-        # Check analog outputs [data with units of %] are floats only
-        columns_to_check = [
-            self.economizer_sig_col,
+        # Set required columns specific to this fault condition
+        self.required_columns = [
+            self.htg_coil_enter_temp_col,
+            self.htg_coil_leave_temp_col,
             self.cooling_sig_col,
             self.heating_sig_col,
+            self.economizer_sig_col,
             self.supply_vfd_speed_col,
         ]
-        self.check_analog_pct(df, columns_to_check)
 
-        # Create helper columns
-        df["htg_delta_temp"] = (
-            df[self.htg_coil_leave_temp_col] - df[self.htg_coil_enter_temp_col]
-        )
+    def get_required_columns(self) -> str:
+        """Returns a string representation of the required columns."""
+        return f"Required columns for FaultConditionFifteen: {', '.join(self.required_columns)}"
+
+    def apply(self, df: pd.DataFrame) -> pd.DataFrame:
+        try:
+            # Ensure all required columns are present
+            self.check_required_columns(df)
+
+            if self.troubleshoot_mode:
+                self.troubleshoot_cols(df)
 
-        df["htg_delta_sqrted"] = (
-            np.sqrt(
-                self.coil_temp_enter_err_thres**2 + self.coil_temp_leav_err_thres**2
+            # Check analog outputs [data with units of %] are floats only
+            columns_to_check = [
+                self.economizer_sig_col,
+                self.cooling_sig_col,
+                self.heating_sig_col,
+                self.supply_vfd_speed_col,
+            ]
+            self.check_analog_pct(df, columns_to_check)
+
+            # Create helper columns
+            df["htg_delta_temp"] = (
+                df[self.htg_coil_leave_temp_col] - df[self.htg_coil_enter_temp_col]
             )
-            + self.delta_supply_fan
-        )
-
-        df["combined_check"] = (
-            (
-                (df["htg_delta_temp"] >= df["htg_delta_sqrted"])
-                # verify AHU is in OS2 only free cooling mode
-                & (df[self.economizer_sig_col] > self.ahu_min_oa_dpr)
-                & (df[self.cooling_sig_col] < 0.1)
+
+            df["htg_delta_sqrted"] = (
+                np.sqrt(
+                    self.coil_temp_enter_err_thres**2 + self.coil_temp_leav_err_thres**2
+                )
+                + self.delta_supply_fan
             )
-            | (
-                (df["htg_delta_temp"] >= df["htg_delta_sqrted"])
-                # OS4 AHU state clg @ min OA
-                & (df[self.cooling_sig_col] > 0.01)
-                & (df[self.economizer_sig_col] == self.ahu_min_oa_dpr)
+
+            df["combined_check"] = (
+                (
+                    (df["htg_delta_temp"] >= df["htg_delta_sqrted"])
+                    # verify AHU is in OS2 only free cooling mode
+                    & (df[self.economizer_sig_col] > self.ahu_min_oa_dpr)
+                    & (df[self.cooling_sig_col] < 0.1)
+                )
+                | (
+                    (df["htg_delta_temp"] >= df["htg_delta_sqrted"])
+                    # OS4 AHU state clg @ min OA
+                    & (df[self.cooling_sig_col] > 0.01)
+                    & (df[self.economizer_sig_col] == self.ahu_min_oa_dpr)
+                )
+                | (
+                    (df["htg_delta_temp"] >= df["htg_delta_sqrted"])
+                    # verify AHU is running in OS 3 clg mode in 100 OA
+                    & (df[self.cooling_sig_col] > 0.01)
+                    & (df[self.economizer_sig_col] > 0.9)
+                )
             )
-            | (
-                (df["htg_delta_temp"] >= df["htg_delta_sqrted"])
-                # verify AHU is running in OS 3 clg mode in 100 OA
-                & (df[self.cooling_sig_col] > 0.01)
-                & (df[self.economizer_sig_col] > 0.9)
+
+            # Rolling sum to count consecutive trues
+            rolling_sum = (
+                df["combined_check"].rolling(window=self.rolling_window_size).sum()
             )
-        )
+            # Set flag to 1 if rolling sum equals the window size
+            df["fc15_flag"] = (rolling_sum >= self.rolling_window_size).astype(int)
 
-        # Rolling sum to count consecutive trues
-        rolling_sum = (
-            df["combined_check"].rolling(window=self.rolling_window_size).sum()
-        )
-        # Set flag to 1 if rolling sum equals the window size
-        df["fc15_flag"] = (rolling_sum >= self.rolling_window_size).astype(int)
+            if self.troubleshoot_mode:
+                print("Troubleshoot mode enabled - not removing helper columns")
+                sys.stdout.flush()
+                del df["htg_delta_temp"]
+                del df["htg_delta_sqrted"]
+                del df["combined_check"]
 
-        if self.troubleshoot_mode:
-            print("Troubleshoot mode enabled - not removing helper columns")
-            sys.stdout.flush()
-            del df["htg_delta_temp"]
-            del df["htg_delta_sqrted"]
-            del df["combined_check"]
+            return df
 
-        return df
+        except MissingColumnError as e:
+            print(f"Error: {e.message}")
+            sys.stdout.flush()
+            raise e  # Re-raise the exception so it can be caught by pytest

open_fdd/air_handling_unit/faults/fault_condition_five.py

@@ -1,7 +1,9 @@
 import pandas as pd
 import pandas.api.types as pdtypes
-from open_fdd.air_handling_unit.faults.fault_condition import FaultCondition
-from open_fdd.air_handling_unit.faults.helper_utils import HelperUtils
+from open_fdd.air_handling_unit.faults.fault_condition import (
+    FaultCondition,
+    MissingColumnError,
+)
 import sys
 
 
@@ -13,6 +15,7 @@ class FaultConditionFive(FaultCondition):
     """
 
     def __init__(self, dict_):
+        super().__init__()
         self.mix_degf_err_thres = float
         self.supply_degf_err_thres = float
         self.delta_t_supply_fan = float
@@ -25,44 +28,60 @@ class FaultConditionFive(FaultCondition):
 
         self.set_attributes(dict_)
 
-    # fault only active if fan is running and htg vlv is modulating
-    # OS 1 is heating mode only fault
+        # Set required columns specific to this fault condition
+        self.required_columns = [
+            self.mat_col,
+            self.sat_col,
+            self.heating_sig_col,
+            self.supply_vfd_speed_col,
+        ]
+
+    def get_required_columns(self) -> str:
+        """Returns a string representation of the required columns."""
+        return f"Required columns for FaultConditionFive: {', '.join(self.required_columns)}"
+
     def apply(self, df: pd.DataFrame) -> pd.DataFrame:
-        if self.troubleshoot_mode:
-            self.troubleshoot_cols(df)
-
-        # check analog outputs [data with units of %] are floats only
-        columns_to_check = [self.supply_vfd_speed_col, self.heating_sig_col]
-
-        for col in columns_to_check:
-            self.check_analog_pct(df, [col])
-
-        df["sat_check"] = df[self.sat_col] + self.supply_degf_err_thres
-        df["mat_check"] = (
-            df[self.mat_col] - self.mix_degf_err_thres + self.delta_t_supply_fan
-        )
-
-        df["combined_check"] = (
-            (df["sat_check"] <= df["mat_check"])
-            # this is to make fault only active in OS1 for htg mode only
-            # and fan is running. Some control programming may use htg
-            # vlv when AHU is off to prevent low limit freeze alarms
-            & (df[self.heating_sig_col] > 0.01)
-            & (df[self.supply_vfd_speed_col] > 0.01)
-        )
-
-        # Rolling sum to count consecutive trues
-        rolling_sum = (
-            df["combined_check"].rolling(window=self.rolling_window_size).sum()
-        )
-        # Set flag to 1 if rolling sum equals the window size
-        df["fc5_flag"] = (rolling_sum == self.rolling_window_size).astype(int)
-
-        if self.troubleshoot_mode:
-            print("Troubleshoot mode enabled - not removing helper columns")
-            sys.stdout.flush()
-            del df["mat_check"]
-            del df["sat_check"]
-            del df["combined_check"]
+        try:
+            # Ensure all required columns are present
+            self.check_required_columns(df)
+
+            if self.troubleshoot_mode:
+                self.troubleshoot_cols(df)
+
+            # Check analog outputs [data with units of %] are floats only
+            columns_to_check = [self.supply_vfd_speed_col, self.heating_sig_col]
+
+            for col in columns_to_check:
+                self.check_analog_pct(df, [col])
 
-        return df
+            df["sat_check"] = df[self.sat_col] + self.supply_degf_err_thres
+            df["mat_check"] = (
+                df[self.mat_col] - self.mix_degf_err_thres + self.delta_t_supply_fan
+            )
+
+            df["combined_check"] = (
+                (df["sat_check"] <= df["mat_check"])
+                & (df[self.heating_sig_col] > 0.01)
+                & (df[self.supply_vfd_speed_col] > 0.01)
+            )
+
+            # Rolling sum to count consecutive trues
+            rolling_sum = (
+                df["combined_check"].rolling(window=self.rolling_window_size).sum()
+            )
+            # Set flag to 1 if rolling sum equals the window size
+            df["fc5_flag"] = (rolling_sum == self.rolling_window_size).astype(int)
+
+            if self.troubleshoot_mode:
+                print("Troubleshoot mode enabled - not removing helper columns")
+                sys.stdout.flush()
+                del df["mat_check"]
+                del df["sat_check"]
+                del df["combined_check"]
+
+            return df
+
+        except MissingColumnError as e:
+            print(f"Error: {e.message}")
+            sys.stdout.flush()
+            raise e  # Re-raise the exception so it can be caught by pytest