summaryrefslogtreecommitdiff
path: root/numpy/fft/tests/test_pocketfft.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/fft/tests/test_pocketfft.py')
-rw-r--r--numpy/fft/tests/test_pocketfft.py14
1 files changed, 4 insertions, 10 deletions
diff --git a/numpy/fft/tests/test_pocketfft.py b/numpy/fft/tests/test_pocketfft.py
index 453e964fa..7c3db0485 100644
--- a/numpy/fft/tests/test_pocketfft.py
+++ b/numpy/fft/tests/test_pocketfft.py
@@ -1,5 +1,3 @@
-from __future__ import division, absolute_import, print_function
-
import numpy as np
import pytest
from numpy.random import random
@@ -7,11 +5,7 @@ from numpy.testing import (
assert_array_equal, assert_raises, assert_allclose
)
import threading
-import sys
-if sys.version_info[0] >= 3:
- import queue
-else:
- import Queue as queue
+import queue
def fft1(x):
@@ -21,13 +15,13 @@ def fft1(x):
return np.sum(x*np.exp(phase), axis=1)
-class TestFFTShift(object):
+class TestFFTShift:
def test_fft_n(self):
assert_raises(ValueError, np.fft.fft, [1, 2, 3], 0)
-class TestFFT1D(object):
+class TestFFT1D:
def test_identity(self):
maxlen = 512
@@ -222,7 +216,7 @@ def test_fft_with_order(dtype, order, fft):
raise ValueError()
-class TestFFTThreadSafe(object):
+class TestFFTThreadSafe:
threads = 16
input_shape = (800, 200)