
    _Mh>                         d dl Z d dlZd dlmZmZ d dlmZ e j        	                    dddd dd dej
        fd	d
d dej        fg          d             Zd Zd Zd Zd Zd Zd Ze j        	                    dg d          d             Zd ZdS )    N)assert_allcloseassert_equal)relative_riskzGexposed_cases, exposed_total, control_cases, control_total, expected_rr)            gUUUUUU?)r   
         r   r
   r   r      c                 X    t          | |||          }t          |j         |d           d S )NgvIh%<=rtol)r   r   )exposed_casesexposed_totalcontrol_casescontrol_totalexpected_rrresults         d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/stats/tests/test_relative_risk.pytest_relative_riskr   	   s:     =-(-9 9FF(+EBBBBBB    c                      t          dddd          } | j         }|                     d          }t          |d           t          |j        |j        fd	d
           d S )N            r   r   r   r   gffffff?)confidence_levelgUUUUUU?)g?ge[@gƠ>r   )r   confidence_intervalr   lowhighr   rrcis      r   &test_relative_risk_confidence_intervalr'      sw    3)+3@ @ @F		B		#	#T	#	:	:B BRVRW%'<4HHHHHHr   c                      t          dddd          } | j         }t          |dd           |                     d	          }t          |j        |j        fd
d           d S )Nr      r      r          @g+=r   r   )r+   r+   g-q=)r   r   r!   r"   r#   r$   s      r    test_relative_risk_ci_conflevel0r,   ,   ss    ")*"> > >F		BB%((((		#	#A	&	&BRVRW%z>>>>>>r   c                      t          dddd          } |                     d          }t          |j        |j        fdt
          j        f           d S )Nr   r)   r   r*   r   r   r   )r   r!   r   r"   r#   npinfr   r&   s     r    test_relative_risk_ci_conflevel1r1   5   sW    ")*"> > >F		#	#A	&	&B"&"'"QK00000r   c                      t          dddd          } t          | j         t          j                   |                                 }t          |j        |j        ft          j        t          j        f           d S )Nr   r)   r*   r   )r   r   r.   nanr!   r"   r#   r0   s     r   #test_relative_risk_ci_edge_cases_00r4   <   m    ")*"> > >F%rv...		#	#	%	%B"&"'"RVRV$455555r   c                      t          dddd          } t          | j         d           |                                 }t          |j        |j        fdt
          j        f           d S )Nr   r)   r   r*   r   g        )r   r   r!   r"   r#   r.   r3   r0   s     r   #test_relative_risk_ci_edge_cases_01r7   D   sh    ")*"> > >F%q)))		#	#	%	%B"&"'"S"&M22222r   c                      t          dddd          } t          | j         t          j                   |                                 }t          |j        |j        ft          j        t          j        f           d S )Nr   r)   r   r*   r   )r   r   r.   r/   r!   r"   r#   r3   r0   s     r   #test_relative_risk_ci_edge_cases_10r9   L   r5   r   zec, et, cc, ct))r   r   r
   r   )r
   r   r   )r   r
   r   r   )r   r
   r:   r   c                     t          j        t          d          5  t          | |||           d d d            d S # 1 swxY w Y   d S )Nz must be an integer not less thanmatch)pytestraises
ValueErrorr   )ecetcccts       r   test_relative_risk_bad_valuerE   T   s    
 
z)K	L	L	L & &b"b"%%%& & & & & & & & & & & & & & & & & &   <A A c                      t          j        t          d          5  t          dddd           d d d            d S # 1 swxY w Y   d S )Nzmust be an integerr<   r   r
   r+   (   )r>   r?   	TypeErrorr    r   r   test_relative_risk_bad_typerK   ]   s    	y(<	=	=	= & &aS"%%%& & & & & & & & & & & & & & & & & &rF   )r>   numpyr.   numpy.testingr   r   scipy.stats.contingencyr   markparametrizer3   r/   r   r'   r,   r1   r4   r7   r9   rE   rK   rJ   r   r   <module>rQ      s]        7 7 7 7 7 7 7 7 1 1 1 1 1 1
 MQBFQBF C C CI I I*? ? ?1 1 16 6 63 3 36 6 6 ) ,< ,< ,< = =& &	= =&
& & & & &r   