
    M/Phq                        d dl Zd dlZd dlmZ d dlmZmZ 	 d dlm	Z
 n# e$ r Y nw xY w ej        d          d             Zej        j        d             Zej        j        d             Zej        j        d	             Zej        j        d
             Zej        j        d             Zej        j        d             Zej        j        d             ZdS )    N)anes96)beanplot
violinplotmodule)scopec                      t          j                    t          j        d          } g d}fd| D             }t          j        |d          }||fS )N   )zStrong DemocratzWeak DemocratzIndependent-DemocratzIndependent-IndependentzIndependent-RepublicanzWeak RepublicanzStrong Republicanc                 H    g | ]}j         d          j        |k             S )age)exogendog).0iddatas     h/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/graphics/tests/test_boxplots.py
<listcomp>z"age_and_labels.<locals>.<listcomp>   s,    
A
A
A"49UDJ",-
A
A
A    object)dtype)r   load_pandasnparangearray)party_IDlabelsr   r   s      @r   age_and_labelsr      sk     Dy||H6 6 6F B
A
A
A
A
A
AC
(3h
'
'
'C;r   c           
      p    | \  }}t          j        dd          \  }}t          |||ddddd           d S N      abssmall   
cutoff_valcutoff_typelabel_fontsizelabel_rotationaxr   	plot_optspltsubplotsr   r   close_figuresr   r   figr*   s         r   test_violinplotr2      s\     KCl1a  GCsr&()%,3,.0 01 1 1 1 1 1r   c                 r    | \  }}t          j        dd          \  }}t          |||dddddd           d S )	Nr   r    r!   r"   r#   g?)r%   r&   r'   r(   	bw_factorr)   r,   r/   s         r   test_violinplot_bw_factorr5   %   s_     KCl1a  GCsr&()%,3,.')+ +, , , , , ,r   c           
      p    | \  }}t          j        dd          \  }}t          |||ddddd           d S r   r-   r.   r   r/   s         r   test_beanplotr8   1   s\     KCl1a  GCSR&'*1*,. ./ / / / / /r   c                 r    | \  }}t          j        dd          \  }}t          |||dddddd           d S )	Nr   Tr    r!   r"   r#   r$   )r*   r   jitterr+   r7   r/   s         r   test_beanplot_jitterr;   <   s^     KCl1a  GCSRt&'*1*,. ./ / / / / /r   c                 t    | \  }}t          j        dd          \  }}t          |||ddddddd	           d S )
Nr   Trightr    r!   r"   r#   r$   r*   r   r:   sider+   r7   r/   s         r   test_beanplot_side_rightr@   G   s`     KCl1a  GCSRt'&'*1*,. ./ / / / / /r   c                 t    | \  }}t          j        dd          \  }}t          |||ddddddd	           d S )
Nr   Tleftr    r!   r"   r#   r$   r>   r7   r/   s         r   test_beanplot_side_leftrC   R   s`     KCl1a  GCSRt&&'*1*,. ./ / / / / /r   c                 j    | \  }}t          j        dd          \  }}t          |||ddi           d S )Nr   bean_legend_texttextr)   r7   r/   s         r   test_beanplot_legend_textrG   ]   sO     KCl1a  GCSR*F35 5 5 5 5 5r   )numpyr   pyteststatsmodels.datasetsr   statsmodels.graphics.boxplotsr   r   matplotlib.pyplotpyplotr-   ImportErrorfixturer   mark
matplotlibr2   r5   r8   r;   r@   rC   rG    r   r   <module>rS      s        ' ' ' ' ' ' > > > > > > > >	####### 	 	 	D	 h	 	  	 1 1 1 , , , / / / / / / / / / / / / 5 5 5 5 5s    ''