smaller diff refactoring

Signed-off-by: Bharath <[email protected]>

Author: bharat-thotakura

cirq-core/cirq/contrib/qasm_import/_parser_test.py

@@ -1014,25 +1014,21 @@ def test_two_qubit_param_gates(
     "qasm_gate", [g[0] for g in two_qubit_gates] + [g[0] for g in two_qubit_param_gates.keys()]
 )
 def test_two_qubit_gates_not_enough_qubits(qasm_gate: str):
-    if qasm_gate in ("cu1", "crz"):
-        qasm = f"""
+    qasm = """
         OPENQASM 2.0;
         include "qelib1.inc";
         qreg q[2];
+    """
+    if qasm_gate in ("cu1", "crz"):
+        qasm += f"""
         {qasm_gate}(0.1) q[0];
     """
     elif qasm_gate == "cu3":
-        qasm = f"""
-        OPENQASM 2.0;
-        include "qelib1.inc";
-        qreg q[2];
+        qasm += f"""
         {qasm_gate}(0.1, 0.2, 0.3) q[0];
     """
     else:
-        qasm = f"""
-        OPENQASM 2.0;
-        include "qelib1.inc";
-        qreg q[2];
+        qasm += f"""
         {qasm_gate} q[0];
     """