blob: 26508f40d5694cd56b42525d76640f513892d579 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#!/usr/bin/env python3
# Copyright 2019 Andrew Myers
#
#
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
# This script tests the "warpx.refine_plasma=1" option by comparing
# the actual number of electrons at step 200 to the expected value
import os
import sys
import yt
yt.funcs.mylog.setLevel(50)
import numpy as np
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import checksumAPI
# this will be the name of the plot file
fn = sys.argv[1]
# Read the file
ds = yt.load(fn)
# count the number of particles
ad = ds.all_data()
np = ad['electrons', 'particle_id'].size
# the number of coarse particle streams
# (odd because one that exactly hits the fine patch is not refined)
n_coarse = 11
# the number of fine particle streams
n_fine = 64
# num particles per stream at time 0
n_0 = 15
# number of times moving window moves
n_move = 99
# ref ratio
rr = 2
np_expected = (n_coarse + n_fine*rr)*(n_0 + n_move)
assert( np == np_expected )
test_name = os.path.split(os.getcwd())[1]
checksumAPI.evaluate_checksum(test_name, fn)
|
