§
    M/PhÒ  ã                   ó‚   — d dl Zd dlmZmZ d dlmZ d dlZd dlm	Z	m
Z
 	 d dlmZ n# e$ r Y nw xY w G d„ d¦  «        ZdS )é    N)Úassert_equalÚassert_raises)ÚSeries)Ú_recodeÚinteraction_plotc                   ól  — e Zd Zed„ ¦   «         Zej        j        d„ ¦   «         Zej        j        d„ ¦   «         Z	ej        j        ej         
                    dddg¦  «        d„ ¦   «         ¦   «         Zej        j        d„ ¦   «         Zej        j        d	„ ¦   «         Zej        j        d
„ ¦   «         Zd„ ZdS )ÚTestInteractionPlotc                 óR  — t           j                             d¦  «         t           j                             ddd¬¦  «        | _        t           j                             ddd¬¦  «        | _        t          j        t           j                             ddd¬¦  «        ¦  «        | _        d S )Ni90  é   é   é<   )Úsizeé   é   )ÚnpÚrandomÚseedÚrandintÚweightÚdurationÚlogÚdays)Úclss    úk/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/graphics/tests/test_factorplots.pyÚsetup_classzTestInteractionPlot.setup_class   s~   € å
Œ	ŠuÑÔÐÝ”Y×&Ò& q¨°Ð&Ñ3Ô3ˆŒ
Ý”y×(Ò(¨¨1°"Ð(Ñ5Ô5ˆŒÝ”6"œ)×+Ò+¨A¨b°rÐ+Ñ:Ô:Ñ;Ô;ˆŒˆˆó    c                 óV   — t          | j        | j        | j        ddgddgd¬¦  «        }d S )NÚredÚblueÚDú^é
   )ÚcolorsÚmarkersÚms©r   r   r   r   ©ÚselfÚclose_figuresÚfigs      r   Útest_plot_bothz"TestInteractionPlot.test_plot_both   s8   € å˜tœ{¨D¬M¸4¼9Ø˜v˜°°S°	¸bðBñ Bô Bˆˆˆr   c                 óP   — t          | j        | j        | j        ddgd¬¦  «        }d S )Nr    r!   r"   ©r$   r%   r&   r'   s      r   Útest_plot_rainbowz%TestInteractionPlot.test_plot_rainbow   s/   € å˜tœ{¨D¬M¸4¼9Ø˜c˜ rð+ñ +ô +ˆˆˆr   ÚastypeÚstrÚintc                 ó  — t          | j        d¬¦  «                             |¦  «        }t          | j        d¬¦  «        }t          | j        d¬¦  «        }t          |||ddgd¬¦  «        }|j        d	         }|                     ¦   «                              ¦   «          	                    ¦   «         }t          |d¦  «         t          |                     ¦   «         d
¦  «         t          |                     ¦   «         d¦  «         d S )NÚWeight)ÚnameÚDurationÚDaysr    r!   r"   r-   r   zmean of Days)r   r   r/   r   r   r   ÚaxesÚ
get_legendÚ	get_titleÚget_textr   Ú
get_ylabelÚ
get_xlabel)	r(   r/   r)   r   r   r   r*   ÚaxÚtraces	            r   Útest_plot_pandasz$TestInteractionPlot.test_plot_pandas!   sç   € õ ˜œ¨(Ð3Ñ3Ô3×:Ò:¸6ÑBÔBˆÝ˜$œ-¨jÐ9Ñ9Ô9ˆÝd”i fÐ-Ñ-Ô-ˆÝ˜v x°Ø(+¨S z°bð:ñ :ô :ˆàŒXaŒ[ˆØ—’‘”×)Ò)Ñ+Ô+×4Ò4Ñ6Ô6ˆÝU˜JÑ'Ô'Ð'ÝR—]’]‘_”_ nÑ5Ô5Ð5ÝR—]’]‘_”_ hÑ/Ô/Ð/Ð/Ð/r   c                 ó¤   — t          | j        | j        | j        ddgddg¬¦  «        }t	          t          |t          j        ¦  «        d¦  «         d S )NÚrÚgú--ú-.)r#   Ú
linestylesT)r   r   r   r   r   Ú
isinstanceÚpltÚFigurer'   s      r   Útest_formattingz#TestInteractionPlot.test_formatting/   sN   € å˜tœ{¨D¬M¸4¼9ÈcÐRUÈYÐdhÐimÐcnÐoÑoÔoˆÝ•Z ¥S¤ZÑ0Ô0°$Ñ7Ô7Ð7Ð7Ð7r   c           	      ó$  — t          t          t          | j        | j        | j        dg¬¦  «         t          t          t          | j        | j        | j        g d¢¬¦  «         t          t          t          | j        | j        | j        g d¢¬¦  «         d S )Nr    )r$   )ÚbrA   rB   )r#   )rC   rD   ú:)rE   )r   Ú
ValueErrorr   r   r   r   )r(   r)   s     r   Útest_formatting_errorsz*TestInteractionPlot.test_formatting_errors4   s†   € å•jÕ"2°D´KÀÄÐPTÔPYÐdgÐchÐiÑiÔiÐiÝ•jÕ"2°D´KÀÄÐPTÔPYÐboÐboÐboÐpÑpÔpÐpÝ•jÕ"2°D´KÀÄÐPTÔPYÐfuÐfuÐfuÐvÑvÔvÐvÐvÐvr   c                 óŠ  — t          | j        | j        | j        d¬¦  «        }t	          t          |t          j        ¦  «        d¦  «         t          | j        | j        | j        d¬¦  «        }t	          t          |t          j        ¦  «        d¦  «         t          t          t           | j        | j        | j        d¬¦  «         d S )NÚline)ÚplottypeTÚscatterÚunknown)
r   r   r   r   r   rF   rG   rH   r   rM   r'   s      r   Útest_plottypez!TestInteractionPlot.test_plottype:   s   € å˜tœ{¨D¬M¸4¼9ÈvÐVÑVÔVˆÝ•Z ¥S¤ZÑ0Ô0°$Ñ7Ô7Ð7Ý˜tœ{¨D¬M¸4¼9ÈyÐYÑYÔYˆÝ•Z ¥S¤ZÑ0Ô0°$Ñ7Ô7Ð7Ý•jÕ"2°D´KÀÄÐPTÔPYÐdmÐnÑnÔnÐnÐnÐnr   c                 óÎ   — t          ddgdz  t          j        ddd¦  «        d¬¦  «        }t          |dd	d
œ¦  «        }t	          |j        j        |j        j        d¬¦  «         d S )NÚarK   r"   r   é(   é   Ú
index_test)Úindexr4   r   )rV   rK   z_recode changed the index)Úerr_msg)r   r   Úaranger   r   rZ   Úvalues)r(   ÚseriesÚseries_s      r   Útest_recode_seriesz&TestInteractionPlot.test_recode_seriesB   s|   € Ý˜˜c˜
 R™­r¬y¸¸BÀÑ/BÔ/BØ)ð+ñ +ô +ˆå˜&¨°Ð"2Ð"2Ñ3Ô3ˆÝW”]Ô)¨6¬<Ô+>Ø8ð	:ñ 	:ô 	:ð 	:ð 	:ð 	:r   N)Ú__name__Ú
__module__Ú__qualname__Úclassmethodr   ÚpytestÚmarkÚ
matplotlibr+   r.   Úparametrizer?   rI   rN   rT   r`   © r   r   r	   r	      s,  € € € € € àð<ð <ñ „[ð<ð „[ÔðBð Bñ ÔðBð „[Ôð+ð +ñ Ôð+ð „[ÔØ„[×Ò˜X¨¨u ~Ñ6Ô6ð
0ð 
0ñ 7Ô6ñ Ôð
0ð „[Ôð8ð 8ñ Ôð8ð „[Ôðwð wñ Ôðwð
 „[Ôðoð oñ Ôðoð:ð :ð :ð :ð :r   r	   )Únumpyr   Únumpy.testingr   r   Úpandasr   re   Ú statsmodels.graphics.factorplotsr   r   Úmatplotlib.pyplotÚpyplotrG   ÚImportErrorr	   ri   r   r   ú<module>rq      sÈ   ðØ Ð Ð Ð Ø 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ø Ð Ð Ð Ð Ð Ø €€€à FÐ FÐ FÐ FÐ FÐ FÐ FÐ Fð	Ø#Ð#Ð#Ð#Ð#Ð#Ð#øØð 	ð 	ð 	Ø€Dð	øøøð9:ð 9:ð 9:ð 9:ð 9:ñ 9:ô 9:ð 9:ð 9:ð 9:s    ' §/®/