
    Ui                     @    d dl Zd dlmZ d dlZd dlmZ d Zd Z	d Z
dS )    N)assert_allclose)minimizec                  4   d } dD ]}t          dd          D ]~}t          | ddd||d          }|                    t          j        d	g                                        d	d	          }|j                                        }t          ||           d S )
Nc                     | dz  d| dz  z  fS )N       )xs    /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/scipy/optimize/tests/test_lbfgsb_hessinv.pyfztest_1.<locals>.f   s    !tQq!tV|    )g:0yE>g-q=g#B;   #   TL-BFGS-B)gtolmaxcor)funjacmethodx0options   )ranger   hess_invnparrayreshapetodenser   )r   r   r   resultH1H2s         r   test_1r"      s       % $ $Bmm 	$ 	$F!jR!%88: : :F 1#//77!<<B((**BB####	$$ $r   c            	         ddgddggfd} t          | dddg	          }t          | d
ddg	          }|j                                        }t          j        |                    t          j        ddg                    |                    t          j        ddg                    f          }t          |                    t          j        ddg                              dd                                        d          |                    t          j        ddg                               t          ||           t          ||j        dd           d S )Nr   r   r      c                     t          j        | t          j        t          j                                      |                     S Nr   dotscipylinalginvr
   H0s    r   r   ztest_2.<locals>.f   1    va 0 0 4 4a88999r   r   
   r   r   r   r   BFGSg{Gz?gQ?)rtolatol)r   r   r   r   vstackr   r   r   )r   result1result2r    r!   r-   s        @r   test_2r8      sj   a&1a&	B: : : : : 1ZRH===G1VR999G			!	!	#	#B	1a&))**1a&))**, 
- 
-B 1a&))11!A6677??CC1a&))**, , , BB(t$??????r   c                      d } ddgddggfd}t          |ddd	g
          }t          |j                                         | |j                             d S )Nc                    | j         | j        | j        | j        f\  }}}}t	          j        | j        d| j        i}|}t          |          D ]}|||         d d t          j	        f         ||         t          j	        d d f         z  ||         z  z
  }|||         d d t          j	        f         ||         t          j	        d d f         z  ||         z  z
  }	t	          j
        |t	          j
        ||	                    ||         ||         d d t          j	        f         z  ||         t          j	        d d f         z  z   }|S )Ndtype)skykn_corrsrhor   eyeshaper;   r   newaxisr(   )
selfsyr>   r?   I_arrHkiA1A2s
             r   todense_old_implz test_3.<locals>.todense_old_impl.   s3   !Wdgt|TXE1gs
5$*55w 	M 	MA1aaam,qtBJM/BBSVKKB1aaam,qtBJM/BBSVKKBBF2rNN++s1v!QQQ
]8K/K89!RZ]8K0L MBB	r   r   r   r   r$   c                     t          j        | t          j        t          j                                      |                     S r&   r'   r,   s    r   r   ztest_3.<locals>.f=   r.   r   r   r/   r   r0   )r   r   r   r   )rK   r   r6   r-   s      @r   test_3rM   ,   s       a&1a&	B: : : : : 1ZRH===GG$,,..0@0@AQ0R0RSSSSSr   )numpyr   numpy.testingr   scipy.linalgr)   scipy.optimizer   r"   r8   rM   r	   r   r   <module>rR      s        ) ) ) ) ) )     # # # # # #$ $ $@ @ @,T T T T Tr   