diff options
| author |  Axel Huebl <axel.huebl@plasma.ninja>
| 2021-12-16 18:58:30 -0800 |
|---|---|---|
| committer |  GitHub <noreply@github.com>
| 2021-12-16 18:58:30 -0800 |
| commit | a194b909f73c44703b013b39dbcead0f278a8ba6 (patch) | |
| tree | fea340813e7ecb633accbd7d50ea92e738faca51 /Examples/Modules/laser_injection/analysis_2d.py | |
| parent | 10d1746419dc9381e657297e03bc255875d5e94d (diff) | |
| download | WarpX-a194b909f73c44703b013b39dbcead0f278a8ba6.tar.gz WarpX-a194b909f73c44703b013b39dbcead0f278a8ba6.tar.zst WarpX-a194b909f73c44703b013b39dbcead0f278a8ba6.zip | |
Generalize Analysis Scripts: Number of Digits (plt) (#2683)
* Update Scripts to Handle New File Names
* Use glob in Analysis of Collision Tests
* Use rstrip in Analysis of Collision Tests
* Regex: Match ends-in-digits
Thanks Phil! :)
Co-authored-by: Edoardo Zoni <ezoni@lbl.gov>
Co-authored-by: Phil Miller <unmobile+gh@gmail.com>
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__": |