aboutsummaryrefslogtreecommitdiff
path: root/Examples/Tests/Langmuir/analysis_langmuir_multi.py
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Tests/Langmuir/analysis_langmuir_multi.py')
-rwxr-xr-xExamples/Tests/Langmuir/analysis_langmuir_multi.py14
1 files changed, 9 insertions, 5 deletions
diff --git a/Examples/Tests/Langmuir/analysis_langmuir_multi.py b/Examples/Tests/Langmuir/analysis_langmuir_multi.py
index 29dbe2255..c45a40219 100755
--- a/Examples/Tests/Langmuir/analysis_langmuir_multi.py
+++ b/Examples/Tests/Langmuir/analysis_langmuir_multi.py
@@ -6,24 +6,28 @@
# 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.rt`. This simulates a 3D periodic plasma wave.
# The electric field in the simulation is given (in theory) by:
# $$ E_x = \epsilon \,\frac{m_e c^2 k_x}{q_e}\sin(k_x x)\cos(k_y y)\cos(k_z z)\sin( \omega_p t)$$
# $$ E_y = \epsilon \,\frac{m_e c^2 k_y}{q_e}\cos(k_x x)\sin(k_y y)\cos(k_z z)\sin( \omega_p t)$$
# $$ E_z = \epsilon \,\frac{m_e c^2 k_z}{q_e}\cos(k_x x)\cos(k_y y)\sin(k_z z)\sin( \omega_p t)$$
-import sys
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 numpy as np
+from scipy.constants import c, e, epsilon_0, m_e
+
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI