Try to have more structured output

lab-cosmo · Nov 16, 2023 · 843a2f9 · 843a2f9
1 parent 9b8d8a5
commit 843a2f9
Showing 1 changed file with 24 additions and 4 deletions.
diff --git a/ipi_tests/test_tools.py b/ipi_tests/test_tools.py
@@ -349,21 +349,41 @@ def run(self, cwd, nid):
 
             # check driver errors
             for driver in drivers:
-                driver_out, driver_err = driver.communicate(timeout=180)
+                driver_out, driver_err = driver.communicate(timeout=60)
                 assert driver.returncode == 0, "DRIVER ERROR OCCURRED: {}".format(
                     driver_err
                 )
 
             # check i-pi errors
-            ipi_out, ipi_error = ipi.communicate(timeout=180)
+            ipi_out, ipi_error = ipi.communicate(timeout=60)
             assert ipi.returncode == 0, "IPI ERROR OCCURRED: {}".format(ipi_error)
 
         except sp.TimeoutExpired:
             print(" TIMEOUT during {} test ", str(cwd))
+
+            try:
+                ipi_out, ipi_error = ipi.communicate(timeout=2)
+            except:
+                ipi_out, ipi_error = "", "Could not get outputs from ipi"
+                pass
+            try:
+                driver_out, driver_err = drivers[0].communicate(timeout=2)                
+            except:
+                driver_out, driver_err = "", "Could not get outputs from drivers"
+                pass
+
             raise RuntimeError(
                 "Time is out. Aborted during {} test. \
-              Error {}".format(
-                    str(cwd), ipi.communicate(timeout=2)[0]
+              **** i-PI output **** \
+              Error {} \
+              Output {} \
+              **** driver output **** \
+              Error {} \
+              Output {} \
+              ".format(
+                    str(cwd), 
+                    ipi_out, ipi_error,
+                    driver_out, driver_err
                 )
             )