3
���h� � @ sH d dl Zd dlZd dlmZmZ d dljZd dl m
Z
G dd� d�Z dS )� N)�Index�
date_range)�cartesian_productc @ sr e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej j
d d d gd d
gd gd
gddgddgdgdgg�dd� �Z dS )�TestCartesianProductc C sj t d�ddg }}t||g�\}}tjddddddg�}tjddddddg�}tj||� tj||� d S )N�ABC� � �A�B�C)�listr �np�array�tm�assert_numpy_array_equal)�self�x�y�result1�result2� expected1� expected2� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/reshape/test_util.py� test_simple
s z TestCartesianProduct.test_simplec C sb t ddd�}dd� t||g�D �\}}tddddg�}tddddg�}tj||� tj||� d S )Nz
2000-01-01� )�periodsc S s g | ]}t |�j�qS r )r �day)�.0r r r r �
<listcomp> s z;TestCartesianProduct.test_datetimeindex.<locals>.<listcomp>r )r r r r �assert_index_equal)r r r r r r r r r �test_datetimeindex s z'TestCartesianProduct.test_datetimeindexc C sF t dddd�}tjddg�}t||g�\}}|jd�}tj||� d S )Nz
2000-01-01r z
US/Pacific)r �tz� � )r r
r r �repeatr r )r r r r r �expectedr r r �test_tzaware_retained s
z*TestCartesianProduct.test_tzaware_retainedc C sL t dddd�jd�}tjddg�}t||g�\}}|jd�}tj||� d S )Nz
2000-01-01r z
US/Pacific)r r" �categoryr# r$ )r Zastyper
r r r% r r )r r r r r r& r r r �!test_tzaware_retained_categorical$ s
z6TestCartesianProduct.test_tzaware_retained_categoricalc C s� g ddgg g}g g dddgg}xjt ||�D ]\\}}tjg tj|�jd�}tjg tj|�jd�}t||g�\}}tj||� tj||� q*W tg �} g }
| |
ks�t�d S )Nr r �a�b�c)�dtype) �zipr
r Zasarrayr- r r r �AssertionError) r �X�Yr r r r r r �resultr& r r r �
test_empty, s z TestCartesianProduct.test_emptyr0 r r r* r+ c
C s, d}t jt|d�� t|d� W d Q R X d S )Nz'Input must be a list-like of list-likes)�match)r0 )�pytestZraises� TypeErrorr )r r0 �msgr r r �test_invalid_input< s z'TestCartesianProduct.test_invalid_inputN) �__name__�
__module__� __qualname__r r! r' r) r3 r5 �markZ parametrizer8 r r r r r s
r ) Znumpyr
r5 Zpandasr r Zpandas._testingZ_testingr Zpandas.core.reshape.utilr r r r r r �<module> s
|