style compliance
@@ -361,7 +361,7 @@ class TestPlanCtxManagerFftn(unittest.TestCase):
try:
with plan_wrong:
- out = fftn(a, s=self.s, axes=self.axes, norm=self.norm)
+ fftn(a, s=self.s, axes=self.axes, norm=self.norm)
except ValueError as ve:
# targeting a particular error
if ve.__str__().startswith('The CUFFT plan and a.shape do'
@@ -438,7 +438,7 @@ class TestPlanCtxManagerFft(unittest.TestCase):
try:
with plan_wrong:
- out = fft(a, n=self.n, norm=self.norm)
+ fft(a, n=self.n, norm=self.norm)
except ValueError as ve:
# targeting a particular error
if ve.__str__().startswith('Target array size does not match'
@@ -449,6 +449,7 @@ class TestPlanCtxManagerFft(unittest.TestCase):
else:
raise Exception("No error is raised --- should not happen.")
+
@testing.parameterize(
{'shape': (3, 4), 's': None, 'axes': None, 'norm': None},
{'shape': (3, 4), 's': None, 'axes': (-2, -1), 'norm': None},
