o
    g                     @   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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)	represent)signc               	   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	t
| d t t dt|   ggks^J d S )Nx   r         r   nqubits)r   r
   ktargetsr   r   r   r   r   r   r   r   r   abs)r    r   [/var/www/visachat/venv/lib/python3.10/site-packages/sympy/physics/quantum/tests/test_qft.pytest_RkGate   s   
2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_represent0   s   
 r5   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   $sympy.functions.elementary.complexesr   r!   r.   r5   r   r   r   r    <module>   s     