aboutsummaryrefslogtreecommitdiff
path: root/Examples/Modules/laser_injection
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Modules/laser_injection')
-rwxr-xr-xExamples/Modules/laser_injection/analysis_1d.py5
-rwxr-xr-xExamples/Modules/laser_injection/analysis_2d.py9
-rwxr-xr-xExamples/Modules/laser_injection/analysis_laser.py5
3 files changed, 14 insertions, 5 deletions
diff --git a/Examples/Modules/laser_injection/analysis_1d.py b/Examples/Modules/laser_injection/analysis_1d.py
index 7c8e985f6..4f4bd505d 100755
--- a/Examples/Modules/laser_injection/analysis_1d.py
+++ b/Examples/Modules/laser_injection/analysis_1d.py
@@ -12,13 +12,16 @@
# the simulation and it compares it with theory. It also checks that the
# central frequency of the Fourier transform is the expected one.
-import yt
import sys
+
import matplotlib
+import yt
+
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import numpy as np
from scipy.signal import hilbert
+
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI
diff --git a/Examples/Modules/laser_injection/analysis_2d.py b/Examples/Modules/laser_injection/analysis_2d.py
index e26ae4478..377931718 100755
--- a/Examples/Modules/laser_injection/analysis_2d.py
+++ b/Examples/Modules/laser_injection/analysis_2d.py
@@ -17,15 +17,18 @@
# the simulation and it compares it with theory. It also checks that the
# central frequency of the Fourier transform is the expected one.
-import yt
+import os
import sys
+
import matplotlib
+
matplotlib.use('Agg')
import matplotlib.pyplot as plt
+from mpl_toolkits.axes_grid1 import make_axes_locatable
import numpy as np
-import os
from scipy.signal import hilbert
-from mpl_toolkits.axes_grid1 import make_axes_locatable
+import yt
+
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI
diff --git a/Examples/Modules/laser_injection/analysis_laser.py b/Examples/Modules/laser_injection/analysis_laser.py
index 5e2234c30..8dde8d6e9 100755
--- a/Examples/Modules/laser_injection/analysis_laser.py
+++ b/Examples/Modules/laser_injection/analysis_laser.py
@@ -8,12 +8,15 @@
# License: BSD-3-Clause-LBNL
+import os
import sys
+
import matplotlib
+
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import numpy as np
-import os
+
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI