
    >qj                        d dl Z d dlmZ d dlmZmZmZmZ i Z eedddg       G d d	             Z	 eeddg
       G d de	             Z
e j                  j                  d ee	dfe       ee
dfe      g      d        Zy)    N)	np_compat)array_namespacemake_xp_pytest_paramxp_assert_closexp_capabilitiesTF)z
dask.array )capabilities_tablecpu_onlyjax_jitskip_backendsc                       e Zd Zd Zd Zd Zy)Ac                 \    t        |      }|| _        t        j                  |      | _        y N)r   _xpnpasarrayxselfr   xps      d/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/scipy/_lib/tests/test_xp_capabilities.py__init__z
A.__init__   s"    QA    c                     t        j                  |      }| j                  j                  t        j                  | j                  |            S r   )r   r   r   matmulr   r   ys     r   fzA.f   s3    JJqMxx		$&&! 455r   c                 H    | j                  |      | j                  |      z   S r   )r   )r   r   zs      r   gzA.g!   s    vvay466!9$$r   N)__name__
__module____qualname__r   r   r"    r   r   r   r      s    
6%r   r   )r	   r
   r   c                       e Zd Zd Zd Zy)Bc                 T    t        |      }|| _        |j                  |      | _        y r   )r   r   r   r   r   s      r   r   z
B.__init__+   s"    QAr   c                 N    | j                   j                  | j                  |      S r   )r   r   r   r   s     r   r   zB.f0   s    xxtvvq))r   N)r#   r$   r%   r   r   r&   r   r   r(   r(   %   s    
*r   r(   clsr"   )r	   c                     |j                  g d      }|j                  g d      }|j                  g d      } | |      }|j                  ||      }|j                  d      d   }t        ||       y )N)g?g@gffffff
@)g      ?g       @      @)r-   g      @g      @g      F@r&   )r   r"   r   )r+   r   r   r   r!   fooobservedexpecteds           r   #test_no_spooky_action_at_a_distancer1   4   sc     	

?#A


?#A


?#A
a&CuuQ{Hzz$#HHh'r   )pytestscipy._lib._array_apir   r   r   r   r   r   local_capabilities_tabler   r(   markparametrizer1   r&   r   r   <module>r7      s     1    /	% %% /%&
* *
* 	aX:RSaX:RS
(
(r   