From e2c0f9dbbcffcf52d67fbd3c3edb43f4c8a7ea4a Mon Sep 17 00:00:00 2001
From: Ting Sun <sunting.05@gmail.com>
Date: Fri, 5 Jul 2024 23:45:28 +0100
Subject: [PATCH] to implement the debug mode Fixes #275

---
 src/supy/_supy_module.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/src/supy/_supy_module.py b/src/supy/_supy_module.py
index 864bdb26..90db99e2 100644
--- a/src/supy/_supy_module.py
+++ b/src/supy/_supy_module.py
@@ -257,6 +257,7 @@ def load_SampleData() -> Tuple[pandas.DataFrame, pandas.DataFrame]:
 
     """
     from ._env import trv_supy_module
+
     trv_SampleData = trv_supy_module / "sample_run"
     path_runcontrol = trv_SampleData / "RunControl.nml"
     df_state_init = init_supy(path_runcontrol, force_reload=False)
@@ -280,7 +281,7 @@ def run_supy(
     logging_level=logging.INFO,
     check_input=False,
     serial_mode=False,
-    debug_mode=False, # TODO: #275 to be implemented to enable debug mode
+    debug_mode=False,  # TODO: #275 to be implemented to enable debug mode
 ) -> Tuple[pandas.DataFrame, pandas.DataFrame]:
     """Perform supy simulation.
 
@@ -348,6 +349,14 @@ def run_supy(
             if isinstance(res_check_state, pd.DataFrame):
                 df_state_init = res_check_state
 
+    # enable debug mode if set
+    if debug_mode:
+        logging_level = logging.DEBUG
+        logger_supy.setLevel(logging_level)
+        df_state_init.loc[:, ("debug", "0")] = 1
+    else:
+        df_state_init.loc[:, ("debug", "0")] = 0
+
     # set up a timer for simulation time
     start = time.time()
 
@@ -376,13 +385,9 @@ def run_supy(
     else:
         logger_supy.info(f"SuPy is running in serial mode")
         try:
-            res_supy = run_supy_ser(
-                df_forcing, df_state_init, save_state, chunk_day
-            )
+            res_supy = run_supy_ser(df_forcing, df_state_init, save_state, chunk_day)
         except:
-            res_supy = run_supy_ser(
-                df_forcing, df_state_init, save_state, chunk_day
-            )
+            res_supy = run_supy_ser(df_forcing, df_state_init, save_state, chunk_day)
 
     # show simulation time
     end = time.time()
@@ -396,7 +401,6 @@ def run_supy(
         return df_output, df_state_final, res_debug
 
 
-
 ##############################################################################
 # 3. save results of a supy run
 def save_supy(