black check

Author: Zichang He

tests/gpumpi_tests.py

@@ -13,6 +13,7 @@
 
 SCRIPT_DIR = Path(__file__).resolve().parent
 
+
 def gpumpi_labs_test(N=10, p=4):
     df = pd.read_json(f"{SCRIPT_DIR.parent}/qokit/assets/QAOA_with_fixed_parameters_p_opt.json", orient="index")
     row = df[(df["N"] == N) & (df["p"] == p)]
@@ -30,26 +31,26 @@ def gpumpi_labs_test(N=10, p=4):
         elif objective == "expectation":
             assert np.isclose(f_gpumpi_labs(gamma, beta), f_c_labs(gamma, beta))
 
+
 def gpumpi_maxcut_test(N=12, p=3, seed=1):
     d = 3
     G = nx.random_regular_graph(d, N, seed=seed)
     gamma, beta = get_fixed_gamma_beta(d, p)
     obj_maxcut = partial(maxcut_obj, w=get_adjacency_matrix(G))
     precomputed_energies = precompute_energies(obj_maxcut, N)
     for objective in ["overlap", "expectation"]:
-        o1_c_maxcut = get_qaoa_maxcut_objective(
-            N, p, precomputed_cuts=precomputed_energies, simulator="c", parameterization="gamma beta", objective=objective
-        )(gamma, beta)
+        o1_c_maxcut = get_qaoa_maxcut_objective(N, p, precomputed_cuts=precomputed_energies, simulator="c", parameterization="gamma beta", objective=objective)(
+            gamma, beta
+        )
         o1_gpumpi_maxcut = get_qaoa_maxcut_objective(
             N, p, precomputed_cuts=precomputed_energies, simulator="gpumpi", parameterization="gamma beta", objective=objective
         )(gamma, beta)
-        o2_gpumpi_maxcut = get_qaoa_maxcut_objective(
-            N, p, G=G, simulator="gpumpi", parameterization="gamma beta", objective=objective
-        )(gamma, beta)
+        o2_gpumpi_maxcut = get_qaoa_maxcut_objective(N, p, G=G, simulator="gpumpi", parameterization="gamma beta", objective=objective)(gamma, beta)
 
-        assert np.isclose(o1_c_maxcut, o1_gpumpi_maxcut.real) 
+        assert np.isclose(o1_c_maxcut, o1_gpumpi_maxcut.real)
         assert np.isclose(o1_c_maxcut, o2_gpumpi_maxcut.real)
 
+
 def gpumpi_sk_test(N=10, p=3, seed=42):
 
     J = get_random_J(N=N, seed=seed)
@@ -63,13 +64,12 @@ def gpumpi_sk_test(N=10, p=3, seed=42):
         o1_gpumpi_sk = get_qaoa_sk_objective(
             N, p, J=J, precomputed_energies=precomputed_energies, simulator="gpumpi", parameterization="gamma beta", objective="overlap"
         )(gamma, beta)
-        o2_gpumpi_sk = get_qaoa_sk_objective(
-            N, p, J=J, simulator="gpumpi", parameterization="gamma beta", objective="overlap"
-        )(gamma, beta)
+        o2_gpumpi_sk = get_qaoa_sk_objective(N, p, J=J, simulator="gpumpi", parameterization="gamma beta", objective="overlap")(gamma, beta)
 
         assert np.isclose(o1_c_sk, o1_gpumpi_sk.real)
         assert np.isclose(o1_c_sk, o2_gpumpi_sk.real)
 
+
 if __name__ == "__main__":
 
     gpumpi_labs_test()

tests/test_gpumpi.py

@@ -33,14 +33,17 @@
 is_gpu = "gpu" in simulators
 SCRIPT_DIR = Path(__file__).resolve().parent
 
+
 @pytest.mark.skipif(not is_nvlink, reason="NVLINK not available.")
 def test_gpumpi(n_procs=n_gpumpi):
     result = subprocess.run(["mpirun", "-np", str(n_procs), "python", f"{SCRIPT_DIR}/gpumpi_tests.py"], capture_output=True, text=True, check=True)
 
+
 @pytest.mark.skipif(not is_gpu, reason="GPU simulator not available.")
 def test_gpumpi_singleprocs():
     result = subprocess.run(["python", f"{SCRIPT_DIR}/gpumpi_tests.py"], capture_output=True, text=True, check=True)
 
+
 @pytest.mark.skipif(not is_gpu, reason="GPU simulator not available.")
 def test_gpu_labs(N=16, p=4, seed=1):
     df = pd.read_json(f"{SCRIPT_DIR.parent}/qokit/assets/QAOA_with_fixed_parameters_p_opt.json", orient="index")