def test_coalescence_call(n_sd, backend, adaptive):
# TODO #330
from PySDM.backends import ThrustRTC
if backend is ThrustRTC:
return
# Arrange
n = np.ones(n_sd)
v = np.ones_like(n)
env = Box(dv=1, dt=default_dt_coal_range[1])
grid = (25, 25)
env.mesh = Mesh(grid, size=grid)
core, sut = get_dummy_core_and_sdm(backend, len(n), environment=env)
cell_id, _, _ = env.mesh.cellular_attributes(
Pseudorandom.sample(grid, len(n)))
attributes = {'n': n, 'volume': v, 'cell id': cell_id}
core.build(attributes)
u01, _ = sut.rnd_opt.get_random_arrays()
sut.actual_length = core.particles._Particles__idx.length
sut.adaptive = adaptive
# Act
sut()
# Assert
np.testing.assert_array_equal(
cell_id, core.particles['cell id'].to_ndarray(raw=True))
def test_coalescence_call(n_sd, backend_class, adaptive):
# TODO #330
if backend_class is ThrustRTC:
return
# Arrange
n = np.ones(n_sd)
v = np.ones_like(n)
env = Box(dv=1, dt=DEFAULTS.dt_coal_range[1])
grid = (25, 25)
env.mesh = Mesh(grid, size=grid)
particulator, sut = get_dummy_particulator_and_coalescence(
backend_class, len(n), environment=env)
cell_id, _, _ = env.mesh.cellular_attributes(
Pseudorandom.sample(grid, len(n)))
attributes = {'n': n, 'volume': v, 'cell id': cell_id}
particulator.build(attributes)
sut.actual_length = particulator.attributes._ParticleAttributes__idx.length
sut.adaptive = adaptive
# Act
sut()
# Assert
np.testing.assert_array_equal(
cell_id, particulator.attributes['cell id'].to_ndarray(raw=True))
