U
|
®ý°dì3 ã @ s® d dl Zd dlZd dlZd dlmZmZmZ d dlm Z
|
dd Z dd Z dd Z ej d ed
|
ed ed g¡d d Zej ded
|
eddfed edfed edfed edfed
|
edfg¡dd Zej dddiddddgid d ig¡d d Zej dd d!g¡ej dd"dd#g¡ej d$d ddd
|
d%d&g¡d'd( Zej d)ed*d+gddgd,ed*d+gd-d.gd,d/fejddgd0d1gd*d+gd2d3d4gd,jejd-d.gd0d1gd*d+gd2d3d4gd,jd5fg¡d6d7 Zej d8d9d:g¡d;d< Zej d=d9d:g¡d>d? Zd@dA ZdBdC Z dDdE Z dFdG Z dHdI Z dJdK Z dLdM Z!dNdO Z"dPdQ Z#dRdS Z$ej dTdUdVg¡dWdX Z%dYdZ Z&d[d\ Z'd]d^ Z(ej dd_d`g¡dadb Z)dS )cé N)Ú CategoricalÚ DataFrameÚSeriesc K s$ t j| |f| t j|| f| dS )a&
|
Check that two Series equal.
|
|
This check is performed commutatively.
|
|
Parameters
|
----------
|
a : Series
|
The first Series to compare.
|
b : Series
|
The second Series to compare.
|
kwargs : dict
|
The arguments passed to `tm.assert_series_equal`.
|
N)ÚtmÚassert_series_equal©ÚaÚbÚkwargs© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/util/test_assert_series_equal.pyÚ_assert_series_equal_both s r c K s> z$t j| |f| d}tj|d W n tk
|
r8 Y nX dS )a
|
Check that two Series are not equal.
|
|
Parameters
|
----------
|
a : Series
|
The first Series to compare.
|
b : Series
|
The second Series to compare.
|
kwargs : dict
|
The arguments passed to `tm.assert_series_equal`.
|
z7The two Series were equal when they shouldn't have been)ÚmsgN)r r ÚpytestZfailÚAssertionError)r r r
|
r r r r Ú_assert_not_series_equal s r c K s t | |f| t || f| dS )a.
|
Check that two Series are not equal.
|
|
This check is performed commutatively.
|
|
Parameters
|
----------
|
a : Series
|
The first Series to compare.
|
b : Series
|
The second Series to compare.
|
kwargs : dict
|
The arguments passed to `tm.assert_series_equal`.
|
N)r r r r r Ú _assert_not_series_equal_both6 s r Údataé Úabcu áà äc C s t t| t| d S ©N)r r )r r r r Útest_series_equalI s r z data1,data2é é Zxyzu éèës aaac C s t t| t| d S r )r r )Údata1Údata2r r r Ú$test_series_not_equal_value_mismatchN s r r
|
ÚdtypeÚfloat64Úindexé ÚnameZfooc C s* t d}t|}t|f| }t|| d S )Nr )Úranger r )r
|
r Ús1Ús2r r r Ú'test_series_not_equal_metadata_mismatch\ s r% )g|ò°Pk¿?g9]¿?)gV-²¿?g Éå?¤¿?Úfloat32ZFloat32Údecimalsé é
|
c C s d| }t | g|d}t |g|d}|dksF|dkrvt| | dkrvd}tjt|d tj|||d W 5 Q R X nt|||d d S ) Nr) ©r )r( r) r gü©ñÒMb@?zSeries values are different©Úmatch©Úrtol)r Úabsr Úraisesr r r r )r r r r' r. r# r$ r r r r Útest_less_precisel s
|
r1 z s1,s2,msgÚl1Úl2©r g ð?g @zSeries\.index are differentgÍÌÌÌÌÌ @g ø?)r r Úcr r z$MultiIndex level \[0\] are differentc C sL d|i}|r8t jt|d tj| |f| W 5 Q R X ntj| |f| d S )NÚcheck_index_typer+ )r r0 r r r )r# r$ r r6 r
|
r r r Ú test_series_equal_index_dtype| s
|
r7 Ú
|
check_likeTFc C sr t dddgdddgd}t dddgdddgd}| s`tjtdd tj||| d
|
W 5 Q R X nt||| d
|
d S ) Nr r r r r r5 r4 úSeries.index are differentr+ )r8 ©r r r0 r r r r )r8 r# r$ r r r Ú test_series_equal_order_mismatch s r; Ú check_indexc C sr t dddgdddgd}t dddgdddgd}| r`tjtdd tj||| d
|
W 5 Q R X nt||| d
|
d S ) Nr r r r r r5 r4 r9 r+ )r< r: )r< r# r$ r r r Ú test_series_equal_index_mismatch§ s r= c C sD t td} t td}tjtdd tj| |ddd W 5 Q R X d S )Nr* z0check_like must be False if check_index is Falser+ FT)r< r8 )r Úobjectr r0 Ú
|
ValueErrorr r ©ÚleftÚrightr r r Ú%test_series_invalid_param_combination³ s
|
|
ÿrC c C sP d}t dddg}t ddddg}tjt|d tj||| d W 5 Q R X d S )NzSeries are different
|
|
Series length are different
|
\[left\]: 3, RangeIndex\(start=0, stop=3, step=1\)
|
\[right\]: 4, RangeIndex\(start=0, stop=4, step=1\)r r r r r+ r- ©r r r0 r r r ©r. r r# r$ r r r Ú!test_series_equal_length_mismatch¼ s
|
rF c C sN d}t dddg}t dddg}tjt|d tj||| d W 5 Q R X d S )NzSeries are different
|
|
Series values are different \(33\.33333 %\)
|
\[index\]: \[0, 1, 2\]
|
\[left\]: \[1, 2, 3\]
|
\[right\]: \[1, 2, 4\]r r r r r+ r- rD rE r r r Ú)test_series_equal_numeric_values_mismatchÊ s
|
rG c C sV d}t tdddg}t tdddg}tjt|d tj||| d W 5 Q R X d S )NzêSeries are different
|
|
Series values are different \(66\.66667 %\)
|
\[index\]: \[0, 1, 2\]
|
\[left\]: \['a', 'b', 'c'\]
|
Categories \(3, object\): \['a', 'b', 'c'\]
|
\[right\]: \['a', 'c', 'b'\]
|
Categories \(3, object\): \['a', 'b', 'c'\]r r r5 r+ r- )r r r r0 r r r rE r r r Ú-test_series_equal_categorical_values_mismatchÙ s
|
rH c C sZ d}t tjdddd}t tjdddd}tjt|d tj||| d W 5 Q R X d S ) NzîSeries are different
|
|
Series values are different \(100.0 %\)
|
\[index\]: \[0, 1, 2\]
|
\[left\]: \[1514764800000000000, 1514851200000000000, 1514937600000000000\]
|
\[right\]: \[1549065600000000000, 1549152000000000000, 1549238400000000000\]z
|
2018-01-01r ÚD)ZperiodsÚfreqz
|
2019-02-02r+ r- )r ÚpdZ
|
date_ranger r0 r r r rE r r r Ú*test_series_equal_datetime_values_mismatchê s
|
rL c C sn d}t tddg}t tddgtdd}| r\tjt|d tj||| d W 5 Q R X nt||| d d S )NzÔAttributes of Series are different
|
|
Attribute "dtype" are different
|
\[left\]: CategoricalDtype\(categories=\['a', 'b'\], ordered=False\)
|
\[right\]: CategoricalDtype\(categories=\['a', 'b', 'c'\], ordered=False\)r r r )Ú
|
categoriesr+ )Úcheck_categorical) r r Úlistr r0 r r r r )rN r r# r$ r r r Ú&test_series_equal_categorical_mismatchù s rP c C sd t tjdddgdd} | t¡}d}tj| |dd tjt |d tj| |d
|
d W 5 Q R X d S ) Nr r r ÚInt64r* zjAttributes of Series are different
|
|
Attribute "dtype" are different
|
\[left\]: Int64
|
\[right\]: int[32|64]F©Z check_dtyper+ T)
|
r rK ÚarrayÚastypeÚintr r r r0 r ©rA rB r r r r Ú1test_assert_series_equal_extension_dtype_mismatch s
|
rW c C sb t t dd¡gdd} | t¡}d}tj| |dd tjt |d tj| |d d W 5 Q R X d S )
|
Nr r Úintervalr* zyAttributes of Series are different
|
|
Attribute "dtype" are different
|
\[left\]: interval\[int64, right\]
|
\[right\]: objectFrR r+ T)
|
r rK ZIntervalrT r> r r r r0 r rV r r r Ú0test_assert_series_equal_interval_dtype_mismatch s
|
rY c C sÌ G dd dt } t ddg}t ddg}| ddg}tj||dd tj||dd tj||dd tj||dd tjtdd tj||dd W 5 Q R X tjtdd tj||dd W 5 Q R X d S )
|
Nc @ s e Zd ZdS )z/test_series_equal_series_type.<locals>.MySeriesN)Ú__name__Ú
|
__module__Ú __qualname__r r r r ÚMySeries. s r] r r F)Zcheck_series_typeTz Series classes are differentr+ ©r r r r r0 r )r] r# r$ Ús3r r r Ú test_series_equal_series_type- s r` c C s¤ t ddg} t ddg}t ddg}tj| |dd tj|| dd d}tjt|d tj| |dd W 5 Q R X d}tjt|d tj|| dd W 5 Q R X d S )Nr r T©Z check_exactzzSeries are different
|
|
Series values are different \(100\.0 %\)
|
\[index\]: \[0, 1\]
|
\[left\]: \[a, b\]
|
\[right\]: \[b, a\]r+ zzSeries are different
|
|
Series values are different \(100\.0 %\)
|
\[index\]: \[0, 1\]
|
\[left\]: \[b, a\]
|
\[right\]: \[a, b\]r^ )r# r$ r_ r r r r Ú&test_series_equal_exact_for_nonnumericB s rb Ú right_dtypeZInt32Úint64c C s8 t dddgdd}t dddg| d}tj||dd d S )Nr r r rQ r* FrR )r r r )rc rA rB r r r Ú8test_assert_series_equal_ignore_extension_dtype_mismatch^ s re c C s¤ t dg} t dgjdd}t | | ¡ t ||¡ tj| |dd tj|| dd tjtdd t | |¡ W 5 Q R X tjtdd t | |¡ W 5 Q R X d S )Nr F)Zallows_duplicate_labels)Z check_flagsz<Flagsr+ )r Z set_flagsr r r r0 r r@ r r r Ú test_allows_duplicate_labelsf s
|
rf c C s< t | g}t || ¡ ¡ t |¡}t ||jdd¡ d S )NT)Údeep)r r r ÚcopyrK ZIndexZassert_index_equal)Z nulls_fixtureZserÚidxr r r Ú%test_assert_series_equal_identical_nau s
|
|
rj c C sp t dddt ddg¡g} t dddt ddg¡g}t | | ¡ tj| | dd t | |¡ tj| |dd d S )Nr g©Ö=¡Û?gß· ¤ò>g ï°´9Þ?g ÓÂýÚ?Tra )r ÚnprS r r )ÚxÚyr r r Ú%test_identical_nested_series_is_equal s$ üÿ üÿ
|
rn Z
|
datetime64Z timedelta64c C sÒ t dddg | d¡}| | d¡}tjtdd t ||¡ W 5 Q R X tj||dd |t d
|
d
|
d
|
g | d¡8 }tjtd d t ||¡ W 5 Q R X tjtd d tj||dd W 5 Q R X d S ) NiC i
iK3Ez[s]z[ms]z"Attributes of Series are differentr+ FrR r zSeries are different)r rT r r0 r r r )r Zser_sZser_msr r r Ú%test_check_dtype_false_different_reso s ro )*Únumpyrk r ZpandasrK r r r Zpandas._testingZ_testingr r r r ÚmarkZ parametrizer" rO r r r% r1 Z from_recordsr5 r7 r; r= rC rF rG rH rL rP rW rY r` rb re rf rj rn ro r r r r Ú<module> s
|
ûþ
|
|
ýþ
|
ý ÿ ÿùøþ
|
|
|
|
|
|
|
