B
    Y                 @   sx   d Z ddlZddlZddlZddlmZ G dd deZ	ej
dddd	 Zej
ddd
d Zej
dddd ZdS )z(Defines chart-wide shared test fixtures.    N)autompgc               @   s    e Zd ZdZdd Zdd ZdS )TestDataz?Contains properties with easy access to data used across tests.c             C   s   ddddg| _ ddddgddddgg| _d	d
 | jD | _| jd | jd d| _t| j| _| jjdd| _t	| _
|   d S )Nacb               c             S   s   g | ]}t |qS  )npZarray).0itemr   r   0lib/python3.7/site-packages/bkcharts/conftest.py
<listcomp>   s    z%TestData.__init__.<locals>.<listcomp>r   )Zcol1Zcol2Zrecords)Zorient)cat_listZ	list_dataZ
array_data	dict_datapd	DataFrameZpd_dataZto_dictZrecords_datar   	auto_data_setup_auto_mpg)selfr   r   r   __init__   s    zTestData.__init__c             C   sR   | j d dk| j d< tj| j d }ttt|jj	 }|
|| j d< d S )NZdispli^  Zlarge_displZcylZreversed_cyl)r   r   ZCategoricalZ
from_arraylistreversedsortedZ
categoriesvaluestolistZreorder_categories)r   catZ	new_orderr   r   r   r      s    zTestData._setup_auto_mpgN)__name__
__module____qualname____doc__r   r   r   r   r   r   r   
   s   r   module)Zscopec               C   s   t  S )N)r   r   r   r   r   	test_data#   s    r%   c             C   s   | j  }| j|d< |S )NZcol3)r   copyr   )r%   datar   r   r   wide_data_with_cat(   s    

r(   c             C   s   t j| j| jdS )N)index)r   r   r   r   )r%   r   r   r   df_with_cat_index/   s    r*   )r#   Znumpyr   Zpandasr   ZpytestZbokeh.sampledata.autompgr   objectr   Zfixturer%   r(   r*   r   r   r   r   <module>   s   