def setUp(self):
"""
Default setup and tests used for 3D NNPS tests
We run the tests on the following pairs of particle arrays:
Set 1) Same particle arrays. Both have constant h.
1) a -> a
2) b -> b
Set 2) Different particle arrays with constant h.
1) a -> b
2) b -> a
Set 3) Variable h
1) c -> c
2) d -> d
We then repeat the above tests again to ensure that we get the
correct results even when running NNPS repeatedly
"""
NNPSTestCase.setUp(self)
self.nps = nnps.DictBoxSortNNPS(
dim=3, particles=self.particles, radius_scale=2.0
)
def setUp(self):
"""Default set-up used by all the tests
Particles with the following coordinates (x, y, z) are placed in a box
0 : -1.5 , 0.25 , 0.5
1 : 0.33 , -0.25, 0.25
2 : 1.25 , -1.25, 1.25
3 : 0.05 , 1.25 , -0.5
4 : -0.5 , 0.5 , -1.25
5 : -0.75, 0.75 , -1.25
6 : -1.25, 0.5 , 0.5
7 : 0.5 , 1.5 , -0.5
8 : 0.5 , -0.5 , 0.5
9 : 0.5 , 1.75 , -0.75
The cell size is set to 1. Valid cell indices and the
particles they contain are given below:
(-2, 0, 0) : particle 0, 6
(0, -1, 0) : particle 1, 8
(1, -2, 1) : particle 2
(0, 1, -1) : particle 3, 7, 9
(-1, 0, -2): particle 4, 5
"""
x = numpy.array([
-1.5, 0.33, 1.25, 0.05, -0.5, -0.75, -1.25, 0.5, 0.5, 0.5])
y = numpy.array([
0.25, -0.25, -1.25, 1.25, 0.5, 0.75, 0.5, 1.5, -0.5, 1.75])
z = numpy.array([
0.5, 0.25, 1.25, -0.5, -1.25, -1.25, 0.5, -0.5, 0.5, -0.75])
# using a degenrate (h=0) array will set cell size to 1 for NNPS
h = numpy.zeros_like(x)
pa = get_particle_array(x=x, y=y, z=z, h=h)
self.dict_box_sort_nnps = nnps.DictBoxSortNNPS(
dim=3, particles=[pa], radius_scale=1.0
)
self.box_sort_nnps = nnps.BoxSortNNPS(
dim=3, particles=[pa], radius_scale=1.0
)
self.ll_nnps = nnps.LinkedListNNPS(
dim=3, particles=[pa], radius_scale=1.0
)
self.sp_hash_nnps = nnps.SpatialHashNNPS(
dim=3, particles=[pa], radius_scale=1.0
)
self.ext_sp_hash_nnps = nnps.ExtendedSpatialHashNNPS(
dim=3, particles=[pa], radius_scale=1.0
)
self.strat_radius_nnps = nnps.StratifiedHashNNPS(
dim=3, particles=[pa], radius_scale=1.0
)
# these are the expected cells
self.expected_cells = {
IntPoint(-2, 0, 0): [0, 6],
IntPoint(0, -1, 0): [1, 8],
IntPoint(1, -2, 1): [2],
IntPoint(0, 1, -1): [3, 7, 9],
IntPoint(-1, 0, -2): [4, 5]
}
def setUp(self):
NNPS2DTestCase.setUp(self)
self.nps = nnps.DictBoxSortNNPS(
dim=2, particles=self.particles, radius_scale=2.0
)
