Remove dependence on QiskitTestCase (#34)

Co-authored-by: Julien Gacon <[email protected]>

Author: woodsp-ibm

test/gradients/test_estimator_gradient.py

@@ -14,6 +14,7 @@
 """Test Estimator Gradients"""
 
 import unittest
+from test import QiskitAlgorithmsTestCase
 
 import numpy as np
 from ddt import ddt, data, unpack
@@ -25,7 +26,6 @@
 from qiskit.primitives import Estimator
 from qiskit.quantum_info import Operator, SparsePauliOp, Pauli
 from qiskit.quantum_info.random import random_pauli_list
-from qiskit.test import QiskitTestCase
 
 from qiskit_algorithms.gradients import (
     FiniteDiffEstimatorGradient,
@@ -49,7 +49,7 @@
 
 
 @ddt
-class TestEstimatorGradient(QiskitTestCase):
+class TestEstimatorGradient(QiskitAlgorithmsTestCase):
     """Test Estimator Gradient"""
 
     @data(*gradient_factories)

test/gradients/test_qfi.py

@@ -14,21 +14,21 @@
 """Test QFI."""
 
 import unittest
-from ddt import ddt, data
+from test import QiskitAlgorithmsTestCase
 
+from ddt import ddt, data
 import numpy as np
 
 from qiskit import QuantumCircuit
 from qiskit.circuit import Parameter
 from qiskit.circuit.parametervector import ParameterVector
 from qiskit.primitives import Estimator
-from qiskit.test import QiskitTestCase
 
 from qiskit_algorithms.gradients import LinCombQGT, ReverseQGT, QFI, DerivativeType
 
 
 @ddt
-class TestQFI(QiskitTestCase):
+class TestQFI(QiskitAlgorithmsTestCase):
     """Test QFI"""
 
     def setUp(self):

test/gradients/test_qgt.py

@@ -14,23 +14,23 @@
 """Test QGT."""
 
 import unittest
-from ddt import ddt, data
+from test import QiskitAlgorithmsTestCase
 
+from ddt import ddt, data
 import numpy as np
 
 from qiskit import QuantumCircuit
 from qiskit.circuit import Parameter
 from qiskit.circuit.library import RealAmplitudes
 from qiskit.primitives import Estimator
-from qiskit.test import QiskitTestCase
 
 from qiskit_algorithms.gradients import DerivativeType, LinCombQGT, ReverseQGT
 
 from .logging_primitives import LoggingEstimator
 
 
 @ddt
-class TestQGT(QiskitTestCase):
+class TestQGT(QiskitAlgorithmsTestCase):
     """Test QGT"""
 
     def setUp(self):

test/gradients/test_sampler_gradient.py

@@ -14,6 +14,7 @@
 """Test Sampler Gradients"""
 
 import unittest
+from test import QiskitAlgorithmsTestCase
 from typing import List
 
 import numpy as np
@@ -25,7 +26,6 @@
 from qiskit.circuit.library.standard_gates import RXXGate
 from qiskit.primitives import Sampler
 from qiskit.result import QuasiDistribution
-from qiskit.test import QiskitTestCase
 
 from qiskit_algorithms.gradients import (
     FiniteDiffSamplerGradient,
@@ -46,7 +46,7 @@
 
 
 @ddt
-class TestSamplerGradient(QiskitTestCase):
+class TestSamplerGradient(QiskitAlgorithmsTestCase):
     """Test Sampler Gradient"""
 
     @data(*gradient_factories)

test/state_fidelities/test_compute_uncompute.py

@@ -13,18 +13,18 @@
 """Tests for Fidelity."""
 
 import unittest
+from test import QiskitAlgorithmsTestCase
 
 import numpy as np
 
 from qiskit.circuit import QuantumCircuit, ParameterVector
 from qiskit.circuit.library import RealAmplitudes
 from qiskit.primitives import Sampler
-from qiskit.test import QiskitTestCase
 
 from qiskit_algorithms.state_fidelities import ComputeUncompute
 
 
-class TestComputeUncompute(QiskitTestCase):
+class TestComputeUncompute(QiskitAlgorithmsTestCase):
     """Test Compute-Uncompute Fidelity class"""
 
     def setUp(self):

test/time_evolvers/test_pvqd.py

@@ -22,7 +22,6 @@
 from qiskit.circuit.library import EfficientSU2
 from qiskit.primitives import Estimator, Sampler
 from qiskit.quantum_info import Pauli, SparsePauliOp
-from qiskit.test import QiskitTestCase
 from qiskit.utils import algorithm_globals
 
 from qiskit_algorithms import AlgorithmError
@@ -270,7 +269,7 @@ def test_aux_ops_raises(self):
             _ = pvqd.evolve(problem)
 
 
-class TestPVQDUtils(QiskitTestCase):
+class TestPVQDUtils(QiskitAlgorithmsTestCase):
     """Test some utility functions for PVQD."""
 
     def setUp(self):