
    !gq                        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
  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y# e$ r Y w xY w)    N)anes96)beanplot
violinplotmodule)scopec                      t        j                         } t        j                  d      }g d}|D cg c]!  }| j                  d   | j
                  |k(     # }}t        j                  |d      }||fS c c}w )N   )zStrong DemocratzWeak DemocratzIndependent-DemocratzIndependent-IndependentzIndependent-RepublicanzWeak RepublicanzStrong Republicanageobject)dtype)r   load_pandasnparangeexogendogarray)dataparty_IDlabelsidr
   s        e/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/graphics/tests/test_boxplots.pyage_and_labelsr      ss     Dyy|H6F 9A
A"499UDJJ",-
AC
A
((3h
'C; Bs   &A5c           
      h    | \  }}t        j                  dd      \  }}t        |||ddddd       y N      abssmall   
cutoff_valcutoff_typelabel_fontsizelabel_rotationaxr   	plot_optspltsubplotsr   r   close_figuresr
   r   figr&   s         r   test_violinplotr.      s=     KCll1a GCsr&()%,3,.01    c                 j    | \  }}t        j                  dd      \  }}t        |||dddddd       y )	Nr   r   r   r   r   g?)r!   r"   r#   r$   	bw_factorr%   r(   r+   s         r   test_violinplot_bw_factorr2   %   s@     KCll1a GCsr&()%,3,.')+,r/   c           
      h    | \  }}t        j                  dd      \  }}t        |||ddddd       y r   r)   r*   r   r+   s         r   test_beanplotr5   1   s=     KCll1a GCSR&'*1*,./r/   c                 j    | \  }}t        j                  dd      \  }}t        |||dddddd       y )	Nr   Tr   r   r   r   r    )r&   r   jitterr'   r4   r+   s         r   test_beanplot_jitterr8   <   s?     KCll1a GCSRt&'*1*,./r/   c                 l    | \  }}t        j                  dd      \  }}t        |||ddddddd	       y )
Nr   Trightr   r   r   r   r    r&   r   r7   sider'   r4   r+   s         r   test_beanplot_side_rightr=   G   sA     KCll1a GCSRt'&'*1*,./r/   c                 l    | \  }}t        j                  dd      \  }}t        |||ddddddd	       y )
Nr   Tleftr   r   r   r   r    r;   r4   r+   s         r   test_beanplot_side_leftr@   R   sA     KCll1a GCSRt&&'*1*,./r/   c                 b    | \  }}t        j                  dd      \  }}t        |||ddi       y )Nr   bean_legend_texttextr%   r4   r+   s         r   test_beanplot_legend_textrD   ]   s5     KCll1a GCSR*F35r/   )numpyr   pyteststatsmodels.datasetsr   statsmodels.graphics.boxplotsr   r   matplotlib.pyplotpyplotr)   ImportErrorfixturer   mark
matplotlibr.   r2   r5   r8   r=   r@   rD    r/   r   <module>rP      s     ' >	#
 h	  	 1 1 , , / / / / / / / / 5 5k  		s   C< <DD