§
    µ³Ùg*  ã                   óÆ   — d dl Zd dlmZmZmZ d dlmZmZm	Z	m
Z
 d dlZd„ Zd„ Zd„ Zd„ Zd„ Zd	„ Zd
„ Zd„ Zej                             dddgddgg¦  «        d„ ¦   «         ZdS )é    N)Úassert_equalÚassert_almost_equalÚassert_allclose)ÚboxcoxÚboxcox1pÚ
inv_boxcoxÚinv_boxcox1pc                  ó   — t          j        g d¢¦  «        } t          | d¦  «        }t          |t          j        | ¦  «        ¦  «         t          | d¦  «        }t          || dz
  ¦  «         t          | d¦  «        }t          |d| dz  dz
  z  ¦  «         t          j        g d¢¦  «        }t          d|¦  «        }t          |d|z  ¦  «         d S )N)ç      à?é   é   é   r   r   r   r   ©r   r   r   ç      ð¿)ÚnpÚarrayr   r   Úlog©ÚxÚyÚlams      úU/granary/edev/quantum/lib/python3.11/site-packages/scipy/special/tests/test_boxcox.pyÚtest_boxcox_basicr   	   sÆ   € Ý
Œ���Ñ Ô €Aõ 	ˆq�!‰Œ€AÝ˜�2œ6 !™9œ9Ñ%Ô%Ð%õ 	ˆq�!‰Œ€AÝ˜˜1˜q™5Ñ!Ô!Ð!õ 	ˆq�!‰Œ€AÝ˜˜3  1¡ q¡™>Ñ*Ô*Ð*õ Œ(�;�;�;Ñ
Ô
€CÝˆq�#‰Œ€AÝ˜˜4 #™:Ñ&Ô&Ð&Ð&Ð&ó    c                  óv   — d} d}t          | |¦  «        }t          |t          j        | ¦  «        d¬¦  «         d S )Ng     ð?ç‘÷P7žxf ç›+¡†›„=©Úrtol)r   r   r   r   ©r   Úlmbdar   s      r   Útest_boxcox_underflowr"      s>   € Ø€AØ€EÝˆq�%ÑÔ€AÝ�A•r”v˜a‘y”y uÐ-Ñ-Ô-Ð-Ð-Ð-r   c                  óx  — t          j        g d¢¦  «        } t          | g d¢¦  «        }t          |t          j        t           j        t           j        t           j        g¦  «        ¦  «         d} t          | ddg¦  «        }t          |t          j        t           j         t           j         g¦  «        ¦  «         d S )N)éÿÿÿÿr$   g      à¿©r   ç       @ç      ø¿r   ç      À)r   r   r   r   ÚnanÚinf©r   r   s     r   Útest_boxcox_nonfiniter,   $   s™   € å
Œ���Ñ Ô €AÝˆqÐ"Ð"Ð"Ñ#Ô#€AÝ�•B”H�bœf¥b¤f­b¬fÐ5Ñ6Ô6Ñ7Ô7Ð7ð 	
€AÝˆq�4˜�)ÑÔ€AÝ�•B”H�rœv˜g­¬ wÐ/Ñ0Ô0Ñ1Ô1Ð1Ð1Ð1r   c                  óš  — t          j        g d¢¦  «        } t          | d¦  «        }t          |t          j        | ¦  «        ¦  «         t          | d¦  «        }t          || ¦  «         t          | d¦  «        }t          |d| z  d| z   z  ¦  «         t          j        g d¢¦  «        }t          d|¦  «        }t          |d|z  ¦  «         d S )	N)g      Ð¿g#B’¡œÇ»r   g#B’¡œÇ;g      Ð?r   é   r   r   r   r   r   r$   r   )r   r   r   r   Úlog1pr   s      r   Útest_boxcox1p_basicr0   0   sÇ   € Ý
ŒÐ6Ð6Ð6Ñ7Ô7€Aõ 	��A‰Œ€AÝ˜�2œ8 A™;œ;Ñ'Ô'Ð'õ 	��A‰Œ€AÝ˜˜1ÑÔÐõ 	��A‰Œ€AÝ˜˜3˜q™5 ! a¡%™=Ñ)Ô)Ð)õ Œ(�;�;�;Ñ
Ô
€CÝ��SÑÔ€AÝ˜˜4 #™:Ñ&Ô&Ð&Ð&Ð&r   c                  óÆ   — t          j        ddg¦  «        } t          j        ddg¦  «        }t          | |¦  «        }t          |t          j        | ¦  «        d¬¦  «         d S )NçVçž¯Ò<r   g¬CÒÑ]r2<r   r   )r   r   r   r   r/   r    s      r   Útest_boxcox1p_underflowr3   E   sZ   € Ý
Œ�%˜�Ñ!Ô!€AÝŒH�f˜e�_Ñ%Ô%€EÝ��EÑÔ€AÝ�A•r”x ‘{”{¨Ð/Ñ/Ô/Ð/Ð/Ð/r   c                  óx  — t          j        g d¢¦  «        } t          | g d¢¦  «        }t          |t          j        t           j        t           j        t           j        g¦  «        ¦  «         d} t          | ddg¦  «        }t          |t          j        t           j         t           j         g¦  «        ¦  «         d S )N)éþÿÿÿr5   r'   r%   r$   r(   r   )r   r   r   r   r)   r*   r+   s     r   Útest_boxcox1p_nonfiniter6   L   s™   € å
Œ���Ñ Ô €AÝ�Ð$Ð$Ð$Ñ%Ô%€AÝ�•B”H�bœf¥b¤f­b¬fÐ5Ñ6Ô6Ñ7Ô7Ð7ð 	€AÝ��T˜1�IÑÔ€AÝ�•B”H�rœv˜g­¬ wÐ/Ñ0Ô0Ñ1Ô1Ð1Ð1Ð1r   c                  óv  — t          j        g d¢¦  «        } t          j        g d¢¦  «        }t          | |¦  «        }t          ||¦  «        }t	          | |¦  «         t          j        g d¢¦  «        } t          j        g d¢¦  «        }t          | |¦  «        }t          ||¦  «        }t	          | |¦  «         d S )N)g        g      ð?r&   )r   r   r   r   r   r   r	   )r   r   r   Úx2s       r   Útest_inv_boxcoxr9   X   s®   € Ý
Œ���ÑÔ€AÝ
Œ(�<�<�<Ñ
 Ô
 €CÝˆq�#‰Œ€AÝ	�A�sÑ	Ô	€BÝ˜˜2ÑÔÐå
Œ���ÑÔ€AÝ
Œ(�<�<�<Ñ
 Ô
 €CÝ��CÑÔ€AÝ	�a˜Ñ	Ô	€BÝ˜˜2ÑÔÐÐÐr   c                  óR   — d} d}t          | |¦  «        }t          || d¬¦  «         d S )Nr2   r   r   r   )r	   r   )r   r   r   s      r   Útest_inv_boxcox1p_underflowr;   f   s6   € Ø€AØ
€CÝ�Q˜ÑÔ€AÝ�A�q˜uÐ%Ñ%Ô%Ð%Ð%Ð%r   zx, lmbéd   é›   g{®Gáz„?ieÿÿÿc                 ó*  — t          | |¦  «        }t          j        |¦  «        sJ ‚t          ||¦  «        }t	          | |¦  «         t          | dz
  |¦  «        }t          j        |¦  «        sJ ‚t          ||¦  «        }t	          | dz
  |¦  «         d S )Nr   )r   r   Úisfiniter   r   r   r	   )r   Úlmbr   Úx_invÚy1pÚx1p_invs         r   Útest_boxcox_premature_overflowrD   m   s˜   € õ 	ˆq�#‰Œ€AÝŒ;�q‰>Œ>ÐÐÐÝ�q˜#ÑÔ€EÝ�A�uÑÔÐõ �1�Q‘3˜Ñ
Ô
€CÝŒ;�sÑÔÐÐÐÝ˜3 Ñ$Ô$€GÝ�A�a‘C˜Ñ!Ô!Ð!Ð!Ð!r   )Únumpyr   Únumpy.testingr   r   r   Úscipy.specialr   r   r   r	   Úpytestr   r"   r,   r0   r3   r6   r9   r;   ÚmarkÚparametrizerD   © r   r   ú<module>rL      s$  ðØ Ð Ð Ð Ø LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LÐ LØ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DØ €€€ð
'ð 'ð 'ð(.ð .ð .ð	2ð 	2ð 	2ð'ð 'ð 'ð*0ð 0ð 0ð	2ð 	2ð 	2ðð ð ð&ð &ð &ð „×ÒØØ	ˆ3€ZØ
ˆD€\ðñô ð
"ð "ñô ð
"ð "ð "r   