added more tests

Author: skushnir123

qhdopt/backend/baseline_backend.py

@@ -4,7 +4,7 @@
 from sympy import lambdify, Symbol, Function
 import jax.numpy as jnp
 
-from qhdopt.backend import dwave_backend, ionq_backend, qutip_backend, baseline_backend
+from qhdopt.backend import dwave_backend, ionq_backend, qutip_backend
 from qhdopt.response import Response
 from qhdopt.utils.function_preprocessing_utils import decompose_function
 
@@ -152,23 +152,6 @@ def qutip_setup(
             gamma=gamma,
         )
 
-    def baseline_setup(
-        self,
-        shots: int = 100,
-    ) -> None:
-        """
-        Sets up a classical baseline backend.
-
-        Args:
-            shots: Number of sampling shots for baseline method.
-        """
-        self.backend = baseline_backend.BaselineBackend(
-            dimension=self.dimension,
-            univariate_dict=self.univariate_dict,
-            bivariate_dict=self.bivariate_dict,
-            shots=shots,
-        )
-
     def compile_only(self):
         self.backend.compile(self.info)
         return self.backend

qhdopt/qhd_base.py

@@ -0,0 +1,24 @@
+import pytest
+
+from qhdopt import QHD
+from qhdopt.backend.dwave_backend import DWaveBackend
+
+def test_backend_performance():
+    Q = [[-2, 1],
+         [1, -1]]
+    bt = [3 / 4, -1 / 4]
+    sol = -.75
+    model = QHD.QP(Q, bt)
+    response = model.classically_optimize(solver="IPOPT", verbose=1)
+    assert abs(response.minimum - sol) < 1e-3
+
+    model.qutip_setup(resolution=6, time_discretization=40)
+    model.optimize(verbose=1)
+    assert abs(response.minimum - sol) < 1e-3
+    model.ionq_setup(resolution=6, api_key='ionq_API_key', time_discretization=5,
+                     shots=1000, on_simulator=True)
+    ionq_backend = model.compile_only()
+    assert len(ionq_backend.qs.evos) == 5
+
+
+