aboutsummaryrefslogtreecommitdiff
path: root/Examples/Modules/space_charge_initialization/analysis.py
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Modules/space_charge_initialization/analysis.py')
-rwxr-xr-xExamples/Modules/space_charge_initialization/analysis.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/Examples/Modules/space_charge_initialization/analysis.py b/Examples/Modules/space_charge_initialization/analysis.py
index 130a54207..0df21e6d9 100755
--- a/Examples/Modules/space_charge_initialization/analysis.py
+++ b/Examples/Modules/space_charge_initialization/analysis.py
@@ -11,15 +11,18 @@ This script checks the space-charge initialization routine, by
verifying that the space-charge field of a Gaussian beam corresponds to
the expected theoretical field.
"""
+import os
import sys
+
import matplotlib
+
matplotlib.use('Agg')
import matplotlib.pyplot as plt
-import yt
import numpy as np
-import os
import scipy.constants as scc
from scipy.special import gammainc
+import yt
+
yt.funcs.mylog.setLevel(0)
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI