test review for input parameters

qiboteam · Sep 9, 2024 · 58200dd · 58200dd
1 parent 30ef970
commit 58200dd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_ui_mpldrawer.py b/tests/test_ui_mpldrawer.py
@@ -21,10 +21,10 @@ def circuit(nqubits=2):
     return c
 
 
-@pytest.mark.parametrize("qubits", [0, 1])
-def test_plot_circuit(qubits):
+@pytest.mark.parametrize("nqubits", [2, 3])
+def test_plot_circuit(nqubits):
     """Test for main plot function"""
-    circ = circuit()
+    circ = circuit(nqubits)
     ax, _ = plot_circuit(circ)
     assert ax.title == ax.title