fix: accept list and ndarray in s and axes

* NumPy fftnd support for list and ndarray for `axes`
* NumPy fftnd support for list and ndarray for `s` when `norm="ortho"`

Author: cako

mkl_fft/_numpy_fft.py

@@ -76,8 +76,8 @@ def frwd_sc_1d(n, s):
 
 
 def frwd_sc_nd(s, axes, x_shape):
-    ss = s if s else x_shape
-    if axes:
+    ss = s if s is not None else x_shape
+    if axes is not None:
         nn = prod([ss[ai] for ai in axes])
     else:
         nn = prod(ss)

mkl_fft/_pydfti.pyx

@@ -932,7 +932,7 @@ def iter_complementary(x, axes, func, kwargs, result):
     nd = x.ndim
     r = list(range(nd))
     sl = [slice(None, None, None)] * nd
-    if not isinstance(axes, tuple):
+    if not np.iterable(axes):
         axes = (axes,)
     for ai in axes:
         r[ai] = None