Address Review Comments

Author: abhishekjiitr

Diff for: quantum/q1.py

@@ -1,28 +1,41 @@
-import qiskit as q
-
+#!/usr/bin/env python3
 """
-Build a simple bare-minimum quantum
-circuit that starts with a single qubit
-(by default in state 0), runs the experiment
-1000 times, and finally prints the total 
-count of the states finally observed.
+Build a simple bare-minimum quantum circuit that starts with a single 
+qubit (by default, in state 0), runs the experiment 1000 times, and 
+finally prints the total count of the states finally observed.
+Qiskit Docs: https://qiskit.org/documentation/getting_started.html
 """
 
-# Use Aer's qasm_simulator
-simulator = q.Aer.get_backend('qasm_simulator')
 
-# Create a Quantum Circuit acting on the q register
-circuit = q.QuantumCircuit(1, 1)
+import qiskit as q
+
+
+def single_qubit_measure() -> q.result.counts.Counts:
+	"""
+	>>> single_qubit_measure()
+	{'0': 1000}
+	"""
+	# Use Aer's qasm_simulator
+	simulator = q.Aer.get_backend('qasm_simulator')
+
+	# Create a Quantum Circuit acting on the q register
+	circuit = q.QuantumCircuit(1, 1)
+
+	# Map the quantum measurement to the classical bits
+	circuit.measure([0], [0])
+
+	# Execute the circuit on the qasm simulator
+	job = q.execute(circuit, simulator, shots=1000)
+
+	# Grab results from the job
+	result = job.result()
 
-# Map the quantum measurement to the classical bits
-circuit.measure([0], [0])
+	# Get the histogram data of an experiment.
+	counts = result.get_counts(circuit)
 
-# Execute the circuit on the qasm simulator
-job = q.execute(circuit, simulator, shots=1000)
+	return counts
 
-# Grab results from the job
-result = job.result()
 
-# Returns counts
-counts = result.get_counts(circuit)
-print("\nTotal count for varopis staes are:", counts)
+if __name__ == '__main__':
+	counts = single_qubit_measure()
+	print("Total count for various states are:", counts)