o
    *ήcE                     @   s   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZmZmZ d dlmZmZmZmZmZmZ d dlmZ d d	lmZ d d
lmZ dd Zdd Zdd ZdS )    )Ipi)Symbol)exp)sqrt)Matrix)QFTIQFTRkGate)ZGateSwapGateHadamardGateCGate	PhaseGateTGate)Qubit)qapply)	representc                  C   s   t d} td| j| ksJ td| jdksJ tddtdks#J tddtdks.J tddtdks9J ttd| ddtddgdt	dt
 t d|   ggksXJ d S )Nx   r         r   nqubits)r   r
   ktargetsr   r   r   r   r   r   r   r   )r    r   K/tmp/pip-target-vg8gfxp4/lib/python/sympy/physics/quantum/tests/test_qft.pytest_RkGate   s   
&r   c                   C   sV  t dd tddtd tdtd td tdtd tdtd td ks/J tdd tdtdtdd tdtdd td tdtdd td tdd ksaJ t	t dddd	t
d
d tdD kswJ t dd sJ tt dd tddd  ttdtd td tddd  ksJ d S )Nr   r   r   )r   r   r   r   c                    s"   g | ]  fd dt dD qS )c                    s4   g | ]}t d t t d |  d  td qS )r      )r   r   r   r   ).0ijr   r   
<listcomp>&   s   4 z3test_quantum_fourier.<locals>.<listcomp>.<listcomp>r"   )range)r#   r   r%   r   r'   &   s   " z(test_quantum_fourier.<locals>.<listcomp>r"      )r   	decomposer   r   r   r   r   r	   r
   r   r   r(   r   r   expandr   r   r   r   test_quantum_fourier   s>   &""r,   c                  C   sF   t dd} t| dd}t|  dd}|jdd|jddks!J d S )Nr   r   r   
   )n)r   r   r*   evalf)cabr   r   r   test_qft_represent.   s   
 r3   N)sympy.core.numbersr   r   sympy.core.symbolr   &sympy.functions.elementary.exponentialr   (sympy.functions.elementary.miscellaneousr   sympy.matrices.denser   sympy.physics.quantum.qftr   r	   r
   sympy.physics.quantum.gater   r   r   r   r   r   sympy.physics.quantum.qubitr   sympy.physics.quantum.qapplyr   sympy.physics.quantum.representr   r   r,   r3   r   r   r   r   <module>   s     