3
h                 @   s  d dl Zd dlZd dlZd dlZd dljZ	d dl
Z
dd Ze
jddddd Ze
jdd	d
d Ze
jdd	dd Ze
j dd Ze
jdd Ze
jdd Ze
jdd Ze
jdd Ze
jdd Ze
jdddgddd Ze
jd d! Ze
jd"d# Ze
jd$d% Ze
jd&d' Ze
jd(d) Ze
jd*d+ Ze
jd,d- Ze
jd.d/ Ze
jd0d1 Ze
jd2d3 Z e
jd4d5 Z!e
jd6d7 Z"dS )8    Nc              C   sN   ddl j} y| jddd}W n
   dS y| jddd}W n
   dS ||kS )z+Helper to verify if Verdana font is presentr   NZVerdanaF)Zfallback_to_defaultZvery_unlikely_to_exist1234T)Zmatplotlib.font_managerZfont_managerZfindfont)ZmplfmZverdana_fontZunlikely_font r   3/tmp/pip-build-riy7u7_k/seaborn/seaborn/conftest.pyhas_verdana
   s    
r   sessionT)Zscopeautousec               C   s@   t jj t jjtj< t jj t jjtj< t jj t jjtj< d S )N)	mpldatesZDateConverterZunitsregistrynpZ
datetime64datetimedater   r   r   r   remove_pandas_unit_conversion!   s    r   )r   c               c   s   d V  t jd d S )Nall)pltcloser   r   r   r   
close_figs-   s    r   c              C   s   t ttd} tjj|  d S )Nzseaborn random global)summapordr
   randomseed)r   r   r   r   random_seed3   s    r   c              C   s   t ttd} tjj| S )Nzseaborn random object)r   r   r   r
   r   ZRandomState)r   r   r   r   rng9   s    r   c             C   sH   t d}tjtjddddd}| jt|t|fd}tj|||dS )	Nabc
   2      Z
wide_index)name)size)indexcolumns)listpd
Int64Indexr
   arangenormallen	DataFrame)r   r    r   valuesr   r   r   wide_df?   s    r)   c             C   s
   t j| S )N)r
   asarray)r)   r   r   r   
wide_arrayH   s    r+   c             C   s.   t jtjdddd}t j| jdd|ddS )Nr      t)r      )r   s)r"   r#   r
   r$   Seriesr%   )r   r   r   r   r   flat_seriesP   s    r1   c             C   s
   t j| S )N)r
   r*   )r1   r   r   r   
flat_arrayW   s    r2   c             C   s   | j  S )N)tolist)r1   r   r   r   	flat_list_   s    r4   seriesarrayr!   )paramsc             C   s   t jtjdddd}t j| jdd|dd}|jdkr>|}nl|jd	krvy|j }W q tk
rr   tj	|}Y qX n4|jd
kry|j
 }W n tk
r   |j }Y nX |S )Nr   r,   r-   )r   r.   )r   r/   r5   r6   r!   )r"   r#   r
   r$   r0   r%   paramZto_numpyAttributeErrorr*   Zto_listr3   )r   requestr   r5   datar   r   r   	flat_datag   s    


r<   c             C   s>   t j| jddtjdddt j| jddtjddddgS )	Nr.   )r   a)r   r         b)r"   r0   r%   r
   r$   )r   r   r   r   wide_list_of_series{   s    rA   c             C   s   dd | D S )Nc             S   s   g | ]}t j|qS r   )r
   r*   ).0r/   r   r   r   
<listcomp>   s    z'wide_list_of_arrays.<locals>.<listcomp>r   )rA   r   r   r   wide_list_of_arrays   s    rD   c             C   s   dd | D S )Nc             S   s   g | ]}|j  qS r   )r3   )rB   r/   r   r   r   rC      s    z&wide_list_of_lists.<locals>.<listcomp>r   )rA   r   r   r   wide_list_of_lists   s    rE   c             C   s   dd | D S )Nc             S   s   i | ]}||j qS r   )r   )rB   r/   r   r   r   
<dictcomp>   s    z'wide_dict_of_series.<locals>.<dictcomp>r   )rA   r   r   r   wide_dict_of_series   s    rG   c             C   s   dd | D S )Nc             S   s   i | ]}t j||jqS r   )r
   r*   r   )rB   r/   r   r   r   rF      s    z'wide_dict_of_arrays.<locals>.<dictcomp>r   )rA   r   r   r   wide_dict_of_arrays   s    rH   c             C   s   dd | D S )Nc             S   s   i | ]}|j  |jqS r   )r3   r   )rB   r/   r   r   r   rF      s    z&wide_dict_of_lists.<locals>.<dictcomp>r   )rA   r   r   r   wide_dict_of_lists   s    rI   c             C   s   d}t jt| jdd|j jd| j|d| j|d| jt	d|| jt	d|| jddg|d	d
g| jt
jdddd|| jdddg|| jdd	g|d	}|d jd}t
j|jjd}|jj||d< |d jd|d< |d jt|d< |S )Nd   r   r.   int)r   r   Zmnop   g333333?gffffff?z
2004-07-30z
2007-07-30zdatetime64[Y])dtyper         g?)	xyzr=   r@   cr-   r/   fr=   categorya_catr/   Zs_catZs_str)r"   r'   dictuniformroundastyper%   Z	lognormalchoicer!   r
   r$   Zrollcat
categoriesZreorder_categoriesstr)r   ndfrV   Znew_categoriesr   r   r   long_df   s"    

ra   c             C   s   | j  S )N)Zto_dict)ra   r   r   r   	long_dict   s    rb   c          	   C   sR   d}t jttjtj|d d| j|d| jtd|tj	tjd|d dS )NrJ   r   )r   r   )rP   rQ   r=   u)
r"   r'   rW   r
   Ztiler$   r%   r[   r!   repeat)r   r_   r   r   r   repeated_df   s    
re   c             C   s>   |j  }x0|D ](}| j|jd d }tj|j||f< qW |S )Nr   )copyZpermutationr   r
   nanloc)r   ra   r`   colidxr   r   r   
missing_df   s
    
rk   c             C   s,   |j  }xdD ]}|| jt||< qW |S )NrS   r/   rT   )rS   r/   rT   )rf   rZ   object)r   ra   r`   ri   r   r   r   	object_df   s    
rm   c             C   s   t j| jddS )NZfloat64)r   rM   )r"   r0   r   )r1   r   r   r   null_series   s    rn   )#Znumpyr
   Zpandasr"   r   Z
matplotlibr   Zmatplotlib.pyplotZpyplotr   Zpytestr   Zfixturer   r   r   r   r)   r+   r1   r2   r4   r<   rA   rD   rE   rG   rH   rI   ra   rb   re   rk   rm   rn   r   r   r   r   <module>   s8   
	

