diff options
Diffstat (limited to 'Examples/Modules/laser_injection/analysis_2d.py')
| -rwxr-xr-x | Examples/Modules/laser_injection/analysis_2d.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/Examples/Modules/laser_injection/analysis_2d.py b/Examples/Modules/laser_injection/analysis_2d.py index 1ea477dbd..e26ae4478 100755 --- a/Examples/Modules/laser_injection/analysis_2d.py +++ b/Examples/Modules/laser_injection/analysis_2d.py @@ -23,6 +23,7 @@ import matplotlib matplotlib.use('Agg') import matplotlib.pyplot as plt import numpy as np +import os from scipy.signal import hilbert from mpl_toolkits.axes_grid1 import make_axes_locatable sys.path.insert(1, '../../../../warpx/Regression/Checksum/') @@ -198,7 +199,7 @@ def main(): check_laser(filename_end) - test_name = filename_end[:-9] # Could also be os.path.split(os.getcwd())[1] + test_name = os.path.split(os.getcwd())[1] checksumAPI.evaluate_checksum(test_name, filename_end) if __name__ == "__main__": |