Python Analysis of Restart Tests: Small Cleanup (#2368)

* Python Analysis of Restart Tests: Small Cleanup

* Remove Unused Import

1 files changed, 1 insertions, 9 deletions

diff --git a/Examples/Tests/restart/analysis_restart.py b/Examples/Tests/restart/analysis_restart.py
index 0c0f01df5..51efc95a8 100755
--- a/Examples/Tests/restart/analysis_restart.py
+++ b/Examples/Tests/restart/analysis_restart.py
@@ -1,7 +1,6 @@
 #! /usr/bin/env python
 
 import sys
-import re
 import yt
 import numpy as np
 sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
@@ -11,9 +10,6 @@ tolerance = sys.float_info.epsilon
 print('tolerance = ', tolerance)
 
 filename = sys.argv[1]
-psatd = True if re.search('psatd', filename) else False
-averaged = True if re.search('avg', filename) else False
-
 ds  = yt.load( filename )
 ad  = ds.all_data()
 xb  = ad['beam',     'particle_position_x'].to_ndarray()
@@ -21,11 +17,7 @@ xe  = ad['plasma_e', 'particle_position_x'].to_ndarray()
 zb  = ad['beam',     'particle_position_z'].to_ndarray()
 ze  = ad['plasma_e', 'particle_position_z'].to_ndarray()
 
-filename = 'orig_restart_plt00010'
-if psatd:
-    filename = 'orig_restart_psatd_plt00010'
-if averaged:
-    filename = 'orig_restart_psatd_time_avg_plt00010'
+filename = 'orig_' + filename
 ds  = yt.load( filename )
 ad  = ds.all_data()
 xb0 = ad['beam',     'particle_position_x'].to_ndarray()