post processing same as festim test

festim-dev · Oct 25, 2023 · a5b4448 · a5b4448
1 parent d254c87
commit a5b4448
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/benchmark.py b/test/benchmark.py
@@ -170,11 +170,12 @@ def siverts_law(T, S_0, E_S, pressure):
     analytical_flux = np.abs(analytical_flux)
     flux_values = np.array(np.abs(flux_values))
 
+    indices = np.where(analytical_flux > 0.01 * np.max(analytical_flux))
+    analytical_flux = analytical_flux[indices]
+    flux_values = flux_values[indices]
+
     relative_error = np.abs((flux_values - analytical_flux) / analytical_flux)
 
-    relative_error = relative_error[
-        np.where(analytical_flux > 0.01 * np.max(analytical_flux))
-    ]
     error = relative_error.mean()
 
     assert error < 0.01