diff --git a/bfps/tools.py b/bfps/tools.py index 9d39e845fcdb6d28e297a775aa874cba6ff9ae39..ee1e4775f560dab3f2ae46e926d45726ca9c7829 100644 --- a/bfps/tools.py +++ b/bfps/tools.py @@ -203,10 +203,11 @@ def padd_with_zeros( """ if (type(odtype) == type(None)): odtype = a.dtype - assert(a.shape[0] <= n0 and - a.shape[1] <= n1 and + print(a.shape, n0, n1, n2//2+1) + assert(a.shape[0] <= n1 and + a.shape[1] <= n0 and a.shape[2] <= n2//2+1) - b = np.zeros((n0, n1, n2//2 + 1) + a.shape[3:], dtype = odtype) + b = np.zeros((n1, n0, n2//2 + 1) + a.shape[3:], dtype = odtype) m0 = a.shape[1] m1 = a.shape[0] m2 = a.shape[2]