finish code coverage

amazon-braket · Dec 27, 2023 · 28aeb32 · 28aeb32
1 parent d563c78
commit 28aeb32
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/braket/circuits/ascii_circuit_diagram.py b/src/braket/circuits/ascii_circuit_diagram.py
@@ -369,19 +369,19 @@ def _ascii_diagram_column(
                     if symbols[qubit] in ["●", "◯"]:
                         if min(target_and_control) < qubit < max(target_and_control):
                             connections[qubit] = "both"
-                        elif qubit < max(target_and_control):
-                            connections[qubit] = "below"
-                        elif min(target_and_control) < qubit:
+                        elif qubit == max(target_and_control):
                             connections[qubit] = "above"
+                        else:
+                            connections[qubit] = "below"
 
                 elif qubit in control_qubits:
                     symbols[qubit] = "●" if map_control_qubit_states[qubit] else "◯"
                     if min(target_and_control) < qubit < max(target_and_control):
                         connections[qubit] = "both"
-                    elif qubit < max(target_and_control):
-                        connections[qubit] = "below"
-                    elif min(target_and_control) < qubit:
+                    elif qubit == max(target_and_control):
                         connections[qubit] = "above"
+                    else:
+                        connections[qubit] = "below"
                 else:
                     symbols[qubit] = "┼"