U
|
ý°d ã @ sJ d dl Zd dlZd dlmZmZ d dlmZ d dl m
|
Z
|
G dd dZ dS )é N)ÚIndexÚ
|
date_range)Úcartesian_productc @ s¬ e Zd Zdd Zdd Zdd Zdd Zej d g g gd
|
d gg gg d d dggg¡dd Z
|
dd Z ej dd d gd dgd gdgd d gd d gd gd gg¡dd Z dd Z dS )ÚTestCartesianProductc C sj t dddg }}t||g\}}t ddddddg¡}t ddddddg¡}t ||¡ t ||¡ d S )NÚABCé é ÚAÚBÚC)Úlistr ÚnpÚarrayÚtmÚassert_numpy_array_equal)ÚselfÚxÚyÚresult1Úresult2Ú expected1Ú expected2© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_util.pyÚ test_simple s z TestCartesianProduct.test_simplec C sn t ddd}dd t||gD \}}tddddgtjd}tddddgtjd}t ||¡ t ||¡ d S )Nú
|
2000-01-01é )Úperiodsc s s | ]}t |jV qd S ©N)r Úday)Ú.0r r r r Ú <genexpr> s z:TestCartesianProduct.test_datetimeindex.<locals>.<genexpr>r ©Údtype)r r r r Úint32r Úassert_index_equal)r r r r r r r r r Útest_datetimeindex s z'TestCartesianProduct.test_datetimeindexc C sF t dddd}t ddg¡}t||g\}}| d¡}t ||¡ d S )Nr r ú
|
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 d¡}t ddg¡}t||g\}}| d¡}t ||¡ d S )Nr r r' r( Úcategoryr* r+ )r Zastyper r r r, r r% r- r r r Ú!test_tzaware_retained_categorical' s
|
|
z6TestCartesianProduct.test_tzaware_retained_categoricalzx, yr r ÚaÚbÚcc C sX t jg t |¡jd}t jg t |¡jd}t||g\}}t ||¡ t ||¡ d S )Nr" )r r Zasarrayr# r r r )r r r r r r r r r r Ú
|
test_empty/ s
|
z TestCartesianProduct.test_emptyc C s t g }g }||kstd S r )r ÚAssertionError)r Úresultr. r r r Útest_empty_input8 s z%TestCartesianProduct.test_empty_inputÚXr c C s, d}t jt|d t|d W 5 Q R X d S )Nz'Input must be a list-like of list-likes©Úmatch©r9 )ÚpytestÚraisesÚ TypeErrorr )r r9 Úmsgr r r Útest_invalid_input> s z'TestCartesianProduct.test_invalid_inputc C sP d}t jt|d4 dd tdD tjdtjdg }t|d W 5 Q R X d S ) Nz+Product space too large to allocate arrays!r: c S s g | ]}t jd dt jdqS )r r r" )r ÚarangeÚint16)r Úir r r Ú
|
<listcomp>L s zBTestCartesianProduct.test_exceed_product_space.<locals>.<listcomp>é i; r" r< )r= r> Ú
|
ValueErrorÚranger rB rC r )r r@ Zdimsr r r Útest_exceed_product_spaceG s ÿz.TestCartesianProduct.test_exceed_product_spaceN)Ú__name__Ú
|
__module__Ú __qualname__r r& r/ r1 r= ÚmarkZ parametrizer5 r8 rA rI r r r r r s
|
(
|
*ÿ
|
r ) Únumpyr r= Zpandasr r Zpandas._testingZ_testingr Zpandas.core.reshape.utilr r r r r r Ú<module> s
|
|
