Address Review Comments

Signed-off-by: Abhishek Jaisingh <[email protected]>

Author: abhishekjiitr

Diff for: quantum/single_qubit_not_gate.py renamed to quantum/not_gate.py

@@ -6,39 +6,34 @@
 Qiskit Docs: https://qiskit.org/documentation/getting_started.html
 """
 
-
 import qiskit as q
 
 
-def single_qubit_measure() -> q.result.counts.Counts:
+def single_qubit_measure(qubits: int, classical_bits: int) -> q.result.counts.Counts:
     """
-    >>> single_qubit_measure()
+    >>> single_qubit_measure(1, 1)
     {'1': 1000}
     """
     # Use Aer's qasm_simulator
-    simulator = q.Aer.get_backend("qasm_simulator")
+    simulator = q.Aer.get_backend('qasm_simulator')
 
     # Create a Quantum Circuit acting on the q register
-    circuit = q.QuantumCircuit(1, 1)
+    circuit = q.QuantumCircuit(qubits, classical_bits)
 
-    # Apply X (NOT) Gate to Qubit 0
+    # Apply X (NOT) Gate to Qubits 0 & 1
     circuit.x(0)
+    circuit.x(1)
 
     # Map the quantum measurement to the classical bits
-    circuit.measure([0], [0])
+    circuit.measure([0, 1], [0, 1])
 
     # Execute the circuit on the qasm simulator
     job = q.execute(circuit, simulator, shots=1000)
 
-    # Grab results from the job
-    result = job.result()
-
-    # Get the histogram data of an experiment.
-    counts = result.get_counts(circuit)
-
-    return counts
+    # Return the histogram data of the results of the experiment.
+    return job.result().get_counts(circuit)
 
 
-if __name__ == "__main__":
-    counts = single_qubit_measure()
-    print("Total count for various states are:", counts)
+if __name__ == '__main__':
+    counts = single_qubit_measure(2, 2)
+    print(f'Total count for various states are: {counts}')