def test_coords(comm): pm = ParticleMesh(BoxSize=8.0, Nmesh=[4, 4, 4], comm=comm, dtype='f8') grid_x = pm.create_coords('real') assert len(grid_x) == 3 assert grid_x[0].dtype == pm.dtype grid_i = pm.create_coords('real', return_indices=True) assert len(grid_i) == 3 grid_x = pm.create_coords('complex') grid_i = pm.create_coords('complex', return_indices=True) assert len(grid_x) == 3 assert grid_x[0].dtype == pm.dtype assert len(grid_i) == 3 pm = ParticleMesh(BoxSize=8.0, Nmesh=[4, 4, 4], comm=comm, dtype='f4') grid_x = pm.create_coords('transposedcomplex') grid_i = pm.create_coords('transposedcomplex', return_indices=True) assert len(grid_x) == 3 assert grid_x[0].dtype == pm.dtype assert len(grid_i) == 3
def test_coords(comm): pm = ParticleMesh(BoxSize=8.0, Nmesh=[4, 4, 4], comm=comm, dtype='f8') grid_x = pm.create_coords('real') assert len(grid_x) == 3 assert grid_x[0].dtype == pm.dtype grid_i = pm.create_coords('real', return_indices=True) assert len(grid_i) == 3 grid_x = pm.create_coords('complex') grid_i = pm.create_coords('complex', return_indices=True) assert len(grid_x) == 3 assert grid_x[0].dtype == pm.dtype assert len(grid_i) == 3 pm = ParticleMesh(BoxSize=8.0, Nmesh=[4, 4, 4], comm=comm, dtype='f4') grid_x = pm.create_coords('transposedcomplex') grid_i = pm.create_coords('transposedcomplex', return_indices=True) assert len(grid_x) == 3 assert grid_x[0].dtype == pm.dtype assert len(grid_i) == 3