aboutsummaryrefslogtreecommitdiff
path: root/Examples/Tests/Langmuir/analysis_langmuir_multi_rz.py
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Tests/Langmuir/analysis_langmuir_multi_rz.py')
-rwxr-xr-xExamples/Tests/Langmuir/analysis_langmuir_multi_rz.py14
1 files changed, 9 insertions, 5 deletions
diff --git a/Examples/Tests/Langmuir/analysis_langmuir_multi_rz.py b/Examples/Tests/Langmuir/analysis_langmuir_multi_rz.py
index d99c6426e..87fd0b367 100755
--- a/Examples/Tests/Langmuir/analysis_langmuir_multi_rz.py
+++ b/Examples/Tests/Langmuir/analysis_langmuir_multi_rz.py
@@ -5,8 +5,7 @@
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
-
-
+#
# This is a script that analyses the simulation results from
# the script `inputs.multi.rz.rt`. This simulates a RZ periodic plasma wave.
# The electric field in the simulation is given (in theory) by:
@@ -14,17 +13,22 @@
# $$ E_z = -\partial_z \phi = - \epsilon \,\frac{mc^2}{e} k_0 \exp\left(-\frac{r^2}{w_0^2}\right) \cos(k_0 z) \sin(\omega_p t)
# Unrelated to the Langmuir waves, we also test the plotfile particle filter function in this
# analysis script.
-import sys
+import os
import re
+import sys
+
import matplotlib
+
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import yt
+
yt.funcs.mylog.setLevel(50)
+
import numpy as np
-import os
-from scipy.constants import e, m_e, epsilon_0, c
import post_processing_utils
+from scipy.constants import c, e, epsilon_0, m_e
+
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI