[testing] [Python] Update test fixture (#3401)

Signed-off-by: Pradnya Khalate <[email protected]>

Author: khalatepradnya

python/tests/parallel/test_mqpu.py

@@ -6,9 +6,15 @@
 # the terms of the Apache License 2.0 which accompanies this distribution.     #
 # ============================================================================ #
 
-import cudaq, os, pytest, random, timeit
-from cudaq import spin
+import os
+import random
+import time
+import timeit
+
+import cudaq
 import numpy as np
+import pytest
+from cudaq import spin
 
 skipIfNoMQPU = pytest.mark.skipif(
     not (cudaq.num_available_gpus() > 0 and cudaq.has_target('nvidia-mqpu')),
@@ -18,6 +24,11 @@
 @pytest.fixture(autouse=True)
 def do_something():
     cudaq.set_target('nvidia-mqpu')
+    try:
+        assert cudaq.get_target().num_qpus(
+        ) > 0, "No QPUs available after target set"
+    except Exception as e:
+        pytest.skip(f"MQPU setup failed: {str(e)}")
     yield
     cudaq.__clearKernelRegistries()
     cudaq.reset_target()
@@ -155,12 +166,6 @@ def parameterized_kernel(n: int, layers: int, params: list[float],
 
 
 def check_accuracy(entity):
-    target = cudaq.get_target()
-    numQpus = target.num_qpus()
-    if numQpus == 0:
-        pytest.skip("No QPUs available for target, skipping test")
-    else:
-        print(f"Target: {target}, NumQPUs: {numQpus}")
     # Define its spin Hamiltonian.
     hamiltonian = 5.907 - 2.1433 * spin.x(0) * spin.x(1) - 2.1433 * spin.y(
         0) * spin.y(1) + .21829 * spin.z(0) - 6.125 * spin.z(1)