diff options
Diffstat (limited to 'Examples/Tests/restart/analysis_restart.py')
-rwxr-xr-x | Examples/Tests/restart/analysis_restart.py | 34 |
1 files changed, 4 insertions, 30 deletions
diff --git a/Examples/Tests/restart/analysis_restart.py b/Examples/Tests/restart/analysis_restart.py index ad79ca969..9ef964664 100755 --- a/Examples/Tests/restart/analysis_restart.py +++ b/Examples/Tests/restart/analysis_restart.py @@ -1,41 +1,15 @@ #! /usr/bin/env python import sys -import yt -import numpy as np sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI -# Load output data generated after restart filename = sys.argv[1] -ds_restart = yt.load(filename) -ds_restart.force_periodicity() -ad_restart = ds_restart.covering_grid(level = 0, - left_edge = ds_restart.domain_left_edge, - dims = ds_restart.domain_dimensions) -# Load output data generated from initial run -benchmark = 'orig_' + filename -ds_benchmark = yt.load(benchmark) -ds_benchmark.force_periodicity() -ad_benchmark = ds_benchmark.covering_grid(level = 0, - left_edge = ds_benchmark.domain_left_edge, - dims = ds_benchmark.domain_dimensions) - -# Loop over all fields (all particle species, all particle attributes, all grid fields) -# and compare output data generated from initial run with output data generated after restart -tolerance = 1e-12 -print('\ntolerance = {:g}'.format(tolerance)) -print() -for field in ds_benchmark.field_list: - dr = ad_restart[field].squeeze().v - db = ad_benchmark[field].squeeze().v - error = np.amax(np.abs(dr - db)) - if (np.amax(np.abs(db)) != 0.): - error /= np.amax(np.abs(db)) - print('field: {}; error = {:g}'.format(field, error)) - assert(error < tolerance) -print() +# Check restart data v. original data +sys.path.insert(0, '../../../../warpx/Examples/') +from analysis_default_restart import check_restart +check_restart(filename) # Check-sum analysis filename = sys.argv[1] |