
    \hS
                         S SK JrJrJrJrJrJrJr  S SKJ	r	  S SK
Jr  S SKJr  S SKJr  S SKJr  S rS rS	 rS
 rS rS rS rg)    )HilbertSpaceComplexSpaceL2	FockSpaceTensorProductHilbertSpaceDirectSumHilbertSpaceTensorPowerHilbertSpace)oo)Symbol)srepr)sstr)Intervalc                      [        5       n [        U [         5      (       d   e[        U 5      S:X  d   e[        U 5      S:X  d   eg )NHzHilbertSpace())r   
isinstancer   r   )hss    `/var/www/auris/envauris/lib/python3.13/site-packages/sympy/physics/quantum/tests/test_hilbert.pytest_hilbert_spacer      s?    	Bb,''''8s??9((((    c                     [        S5      n [        U [         5      (       d   eU R                  S:X  d   e[        U 5      S:X  d   e[	        U 5      S:X  d   e[        S5      n[        U5      n[        U[         5      (       d   eUR                  U:X  d   e[        U5      S:X  d   e[	        U5      S:X  d   eUR                  US5      [        S5      :X  d   eg )N   zC(2)zComplexSpace(Integer(2))nzC(n)zComplexSpace(Symbol('n')))r   r   	dimensionr   r   r   subs)c1r   c2s      r   test_complex_spacer      s    	aBb,''''<<18v92222sA	aBb,''''<<18v93333771a=LO+++r   c                     [        [        [        * S5      5      n [        U [         5      (       d   eU R                  [        L d   eU R
                  [        [        * S5      :X  d   e[        SSS9n[        SSS9n[        [        X5      5      nUR                  [        L d   eUR
                  [        X5      :X  d   eUR                  US5      [        [        SU5      5      :X  d   eg )N   xT)realy)r   r   r
   r   r   intervalr   r   )b1r    r"   b2s       r   test_L2r'   $   s    	HbS!	Bb"<<2;;(B3****sAsA	HQN	B<<2;;(1.(((771b>RQ0000r   c                      [        5       n [        5       n[        U [         5      (       d   eU R                  [        L d   eX:X  d   eg )N)r   r   r   r
   )f1f2s     r   test_fock_spacer+   2   s>    	B	Bb)$$$$<<28O8r   c                     [        S5      n [        S5      n[        U 5      nX-  n[        U[        5      (       d   eUR                  SU -  :X  d   eUR
                  X4:X  d   eX"-  n[        U[        5      (       d   eUR                  U:X  d   eUR                  S:X  d   eUR                  U S-  :X  d   e[        5       nX-  U-  nUR                  [        L d   eg Nr   r   )r   r   r   r   r   spacesr	   baseexpr   r
   r   hs1hs2hfs        r   test_tensor_productr6   :   s    sA
q/C
q/CAa23333;;!A#88z!!!Aa0111166S==55A::;;!Q$A	A;;"r   c                     [        S5      n [        S5      n[        U 5      nUS-  n[        U[        5      (       d   eUR                  U:X  d   eUR
                  S:X  d   eUR                  S:X  d   eUS-  n[        U[        5      (       d   eUR                  U:X  d   eUR
                  S:X  d   eUR                  U S-  :X  d   eg )Nr   r         )r   r   r   r	   r/   r0   r   )r   r2   r3   r4   s       r   test_tensor_powerr:   O   s    sA
q/C
q/CQAa0111166S==55A::;;!QAa0111166S==55A::;;!Q$r   c                  @   [        S5      n [        S5      n[        U 5      nX-   n[        U[        5      (       d   eUR                  SU -   :X  d   eUR
                  X4:X  d   e[        5       nX-   U-   nUR                  [        L d   eUR
                  XU4:X  d   eg r-   )r   r   r   r   r   r.   r   r
   r1   s        r   test_direct_sumr<   a   s    sA
q/C
q/C	Aa.////;;!a%88z!!!A#A;;"88}$$$r   N)sympy.physics.quantum.hilbertr   r   r   r   r   r   r	   sympy.core.numbersr
   sympy.core.symbolr   sympy.printing.reprr   sympy.printing.strr   sympy.sets.setsr   r   r   r'   r+   r6   r:   r<    r   r   <module>rD      sH     
 " $ % # $), 1*$%r   