diff options
Diffstat (limited to 'Examples/Modules/space_charge_initialization/analysis.py')
| -rwxr-xr-x | Examples/Modules/space_charge_initialization/analysis.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/Examples/Modules/space_charge_initialization/analysis.py b/Examples/Modules/space_charge_initialization/analysis.py index 30954c41d..c84962e89 100755 --- a/Examples/Modules/space_charge_initialization/analysis.py +++ b/Examples/Modules/space_charge_initialization/analysis.py @@ -78,8 +78,12 @@ plt.colorbar() plt.savefig('Comparison.png') # Automatically check the results -print( 'Relative error in Ex: %.3f'%(abs(Ex_array-Ex_th).max()/Ex_th.max())) -assert np.allclose( Ex_array, Ex_th, atol=0.15*Ex_th.max() ) -assert np.allclose( Ey_array, Ey_th, atol=0.15*Ey_th.max() ) +def check(E, E_th, label): + print( 'Relative error in %s: %.3f'%( + label, abs(E-E_th).max()/E_th.max())) + assert np.allclose( E, E_th, atol=0.15*E_th.max() ) + +check( Ex_array, Ex_th, 'Ex' ) +check( Ey_array, Ey_th, 'Ey' ) if ds.dimensionality == 3: - assert np.allclose( Ez_array, Ez_th, atol=0.15*Ez_th.max() ) + check( Ez_array, Ez_th, 'Ez' ) |