Coverage for pySDC/implementations/problem_classes/boussinesq_helpers/unflatten.py: 0%

1import numpy as np 

2 

3 

4def unflatten(uin, dim, Nx, Nz): 

5 temp = np.asarray(np.split(uin, dim)) 

6 uout = np.zeros((dim, Nx, Nz)) 

7 for i in range(0, dim): 

8 uout[i, :, :] = np.split(temp[i, :], Nx) 

9 return uout