Skip to content

Commit 2c88e22

Browse files
Fix doctest q fourier transform.py #9943 (#10624)
* Fixed failed examples * Added testmod function * Revert pre-commit changes Co-authored-by: Christian Clauss <cclauss@me.com> * Add print statements for quantum Fourier transform output Added print statements to display the total count for the quantum Fourier transform state when the module is run directly. --------- Co-authored-by: Christian Clauss <cclauss@me.com>
1 parent 14c971f commit 2c88e22

1 file changed

Lines changed: 25 additions & 5 deletions

File tree

quantum/q_fourier_transform.py

Lines changed: 25 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,26 @@ def quantum_fourier_transform(number_of_qubits: int = 3) -> qiskit.result.counts
6767
>>> quantum_fourier_transform(0.5)
6868
Traceback (most recent call last):
6969
...
70-
ValueError: number of qubits must be exact integer.
70+
ValueError: number of qubits must be an exact integer.
71+
72+
>>> result = quantum_fourier_transform(2)
73+
>>> 2350<=result['10']<=2600
74+
True
75+
>>> 2350<=result['00']<=2600
76+
True
77+
>>> 2350<=result['11']<=2600
78+
True
79+
>>> 2350<=result['01']<=2600
80+
True
81+
>>> res = quantum_fourier_transform(3)
82+
>>> 1150<=res['000']<=1350 and 1150<=res['001']<=1350
83+
True
84+
>>> 1150<=res['010']<=1350 and 1150<=res['100']<=1350
85+
True
86+
>>> 1150<=res['101']<=1350 and 1150<=res['110']<=1350
87+
True
88+
>>> 1150<=res['011']<=1350 and 1150<=res['111']<=1350
89+
True
7190
"""
7291
if isinstance(number_of_qubits, str):
7392
raise TypeError("number of qubits must be a integer.")
@@ -107,7 +126,8 @@ def quantum_fourier_transform(number_of_qubits: int = 3) -> qiskit.result.counts
107126

108127

109128
if __name__ == "__main__":
110-
print(
111-
f"Total count for quantum fourier transform state is: \
112-
{quantum_fourier_transform(3)}"
113-
)
129+
import doctest
130+
131+
doctest.testmod()
132+
print("Total count for quantum Fourier transform state is:")
133+
print(f"{quantum_fourier_transform(3) = }")

0 commit comments

Comments
 (0)