Remove unused import

qutech · Jul 3, 2020 · 0fd65de · 0fd65de
1 parent a6afbb9
commit 0fd65de
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/filter_functions/plotting.py b/filter_functions/plotting.py
@@ -192,8 +192,6 @@ def plot_bloch_vector_evolution(pulse: 'PulseSequence',
     --------
     qutip.bloch.Bloch: Qutip's Bloch sphere implementation.
     """
-    from mpl_toolkits.mplot3d import Axes3D
-
     # Raise an exception if not a one-qubit pulse
     if not pulse.d == 2:
         raise ValueError('Plotting Bloch sphere evolution only implemented ' +

diff --git a/tests/test_extras.py b/tests/test_extras.py
@@ -55,8 +55,7 @@ def test_plotting_not_available(self):
         reason='Skipping tests for missing bloch sphere visualization tests in build with qutip')  # noqa
     def test_bloch_sphere_visualization_not_available(self):
 
-        with self.assertWarns(UserWarning):
-            from filter_functions import plotting
+        from filter_functions import plotting
 
         with self.assertRaises(RuntimeError):
             plotting.get_bloch_vector(testutil.rng.standard_normal((10, 2)))

diff --git a/tests/testutil.py b/tests/testutil.py
@@ -26,14 +26,14 @@
 from pathlib import Path
 
 import numpy as np
-from numpy.random import RandomState
+from numpy import random
 from numpy.testing import assert_allclose, assert_array_equal
 from scipy import io
 from scipy import linalg as sla
 
 from filter_functions import Basis, PulseSequence, util
 
-rng = RandomState()
+rng = random.RandomState()
 
 
 class TestCase(unittest.TestCase):