aboutsummaryrefslogtreecommitdiff
path: root/Examples/Tests/restart/analysis_restart.py
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Tests/restart/analysis_restart.py')
-rwxr-xr-xExamples/Tests/restart/analysis_restart.py34
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]