add a bit more tests
@@ -378,6 +378,12 @@ class TestFftAllocate:
{'shape': (3, 4), 's': None, 'axes': (-2, -1), 'norm': None},
{'shape': (3, 4), 's': None, 'axes': (-1, -2), 'norm': None},
{'shape': (3, 4), 's': None, 'axes': (0,), 'norm': None},
+ {'shape': (3, 4), 's': None, 'axes': None, 'norm': 'forward'},
+ {'shape': (3, 4), 's': (1, None), 'axes': None, 'norm': 'forward'},
+ {'shape': (3, 4), 's': (1, 5), 'axes': None, 'norm': 'forward'},
+ {'shape': (3, 4), 's': None, 'axes': (-2, -1), 'norm': 'forward'},
+ {'shape': (3, 4), 's': None, 'axes': (-1, -2), 'norm': 'forward'},
+ {'shape': (3, 4), 's': None, 'axes': (0,), 'norm': 'forward'},
{'shape': (3, 4), 's': None, 'axes': None, 'norm': 'ortho'},
{'shape': (3, 4), 's': None, 'axes': (), 'norm': None},
{'shape': (2, 3, 4), 's': None, 'axes': None, 'norm': None},
@@ -386,9 +392,16 @@ class TestFftAllocate:
{'shape': (2, 3, 4), 's': None, 'axes': (-3, -2, -1), 'norm': None},
{'shape': (2, 3, 4), 's': None, 'axes': (-1, -2, -3), 'norm': None},
{'shape': (2, 3, 4), 's': None, 'axes': (0, 1), 'norm': None},
+ {'shape': (2, 3, 4), 's': None, 'axes': None, 'norm': 'forward'},
+ {'shape': (2, 3, 4), 's': (1, 4, None), 'axes': None, 'norm': 'forward'},
+ {'shape': (2, 3, 4), 's': (1, 4, 10), 'axes': None, 'norm': 'forward'},
+ {'shape': (2, 3, 4), 's': None, 'axes': (-3, -2, -1), 'norm': 'forward'},
+ {'shape': (2, 3, 4), 's': None, 'axes': (-1, -2, -3), 'norm': 'forward'},
+ {'shape': (2, 3, 4), 's': None, 'axes': (0, 1), 'norm': 'forward'},
{'shape': (2, 3, 4), 's': None, 'axes': None, 'norm': 'ortho'},
{'shape': (2, 3, 4), 's': None, 'axes': (), 'norm': None},
{'shape': (2, 3, 4), 's': (2, 3), 'axes': (0, 1, 2), 'norm': 'ortho'},
+ {'shape': (2, 3, 4), 's': (2, 3), 'axes': (0, 1, 2), 'norm': 'forward'},
{'shape': (2, 3, 4, 5), 's': None, 'axes': None, 'norm': None},
{'shape': (0, 5), 's': None, 'axes': None, 'norm': None},
{'shape': (2, 0, 5), 's': None, 'axes': None, 'norm': None},
