a
    kh                     @   s|   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 d dlmZmZmZmZ dd	 Zd
d ZdS )    )prod)Rational)exp)sqrt)Dagger
Commutatorqapply)BosonOp)BosonFockKetBosonFockBraBosonCoherentKetBosonCoherentBrac                  C   s   t d} t d}t| t sJ tt| t s0J | js:J t| jrHJ t dt ddks^J t dt dksrJ t ddt ddksJ t| t|  dksJ t| t| | t| t||   ksJ tt| tt| ksJ d S )NabTcF   )r	   
isinstancer   Zis_annihilationr   doitr   )r   r    r   T/var/www/auris/lib/python3.9/site-packages/sympy/physics/quantum/tests/test_boson.pytest_bosonoperator   s    
.r   c                  C   s  t d} d}tdtd  dks(J tdtd  dksDJ tt|t| |  td tttd|d ks~J d\}}t	|t
|  dksJ t	|t
|  dksJ tt	|t
|  t|| d tdd  dk sJ t| t
| |t
| ksJ d S )	Nr      r   r   )g333333?g333333@   g-q=)r	   r   r
   r   r   r   r   r   ranger   r   absr   r   )r   nZalpha1Zalpha2r   r   r   test_boson_states!   s$    
r   N)mathr   Zsympy.core.numbersr   Z&sympy.functions.elementary.exponentialr   Z(sympy.functions.elementary.miscellaneousr   Zsympy.physics.quantumr   r   r   Zsympy.physics.quantum.bosonr	   r
   r   r   r   r   r   r   r   r   r   <module>   s   