3
Ud                 @   s   d dl Zd dlZd dlZd dljZejj	dd dddgdd Z
ejj	ddddgdd Zdd Zdd Zdd Zdd ZdS )    N
align_axis   indexcolumnsc             C   s   t jdddg}t jdddg}|j|| d}| dkr|t jd	d
g}t jddg}t jddgddgg||d}tj|| n:t jjd	d
gddgg}t jddddg|d}tj	|| d S )Nabcxz)r   r   r   r      selfother)r   r   )r   )r   r   )
pdSeriescompareIndex	DataFrametmassert_frame_equal
MultiIndexfrom_productassert_series_equal)r   s1s2resultindicesr   expected r   \/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/series/methods/test_compare.pytest_compare_axis   s    r   zkeep_shape, keep_equalTFc             C   s   t jdddg}t jdddg}|j|| |d}| rt jddd	g}t jd
dg}|rzt jddgddgddgg||d}qt jddgtjtjgddgg||d}n8t jdd	g}t jd
dg}t jddgddgg||d}tj|| d S )Nr   r   r   r	   r
   )
keep_shape
keep_equalr   r   r   r   r   )r   r   )	r   r   r   r   r   npnanr   r   )r    r!   r   r   r   r   r   r   r   r   r   test_compare_various_formats   s$    
 
r$   c              C   sV   t jddtjg} t jddtjg}| j|}t jddggddgd}tj|| d S )Nr   r   r	   r   r   )r   )r   r   r"   r#   r   r   r   r   )r   r   r   r   r   r   r   test_compare_with_equal_nullsB   s
    
r%   c              C   sp   t jdddg} t jddtjg}| j|dd}t jjddgdd	gg}t jdddtjg|d
}tj|| d S )Nr   r   r   r	   r   )r   r   r   r   )r   )	r   r   r"   r#   r   r   r   r   r   )r   r   r   r   r   r   r   r   !test_compare_with_non_equal_nullsM   s    r&   c              C   s   t jjdddgdddgg} t jdddg| d}t jddd	g| d}|j|dd
}t jjddddgddddgddddgg}t jdddd	g|d}tj|| d S )Nr   r   r   r   r   r   )r   r	   r
   )r   r   r   )r   r   Zfrom_arraysr   r   r   r   )r   r   r   r   r   r   r   r   r   test_compare_multi_indexY   s    $r'   c              C   s   d} t jt| dD tjdddgdddgd	}tjdddgddd
gd	}|j| W d Q R X d} t jt| d2 tjdddg}tjddddg}|j| W d Q R X d S )Nz3Can only compare identically-labeled Series objects)matchr   r      r   r   r   )r   d   )pytestZraises
ValueErrorr   r   r   )msgZser1Zser2r   r   r   test_compare_unaligned_objectsg   s    r/   )TF)FT)TT)Znumpyr"   r,   Zpandasr   Zpandas._testingZ_testingr   markZparametrizer   r$   r%   r&   r'   r/   r   r   r   r   <module>   s   
 