B x\ @sddlZddlZddlmZddlZddlmm Z ddZ ej dej dejdgddd Zd d ejej gDZed d ejejejgDedd edgej edddZej eejdddeejdddeejdddedgdddddZej edededgdddddZej ej !dej "dejdded e#dd!e#dd"gd#ddd$d%Z$ej ej "d&ej %d'ejd&d(ejd)d*e#d&d"e#d'd+gd,ddd-d.Z&ej 'd&ej (dej %gZ)ej ejd/d0e#d/d1ejd/d*ge)dd2d3Z*ej e#d4d"ejd5d(ed6ge)dd7d8Z+ej e#d9d!ejd9dejd9dge)dd:d;Z,ej ejej-ej.ge ddge dd?d@Z3ej ejdAfej-dAfej.dAfej0ej.dBfej1j2d>ge ddCdDZ4ej ejej-ej.e j5ge ddEdFZ6e6Z7dS)GN)longcCs>t|tr4t|dkst|djdt|dS|jSdS)Nr-) isinstancetuplelenAssertionError__name__str)xr ?lib/python3.7/site-packages/pandas/tests/arithmetic/conftest.pyid_funcs rr)dtype)paramscCs|jS)N)param)requestr r ronesrcCs4g|],}tjtjtjgD]}|dgd|dqqS)r)r)npint64uint64float64).0Zbox_clsrr r r srcCsg|]}tjd|dqS)r)r)rarray)rrr r rr!sgcCs|jS)N)r)rr r rzero&srrrrrcCs t|jS)N)typer )r r r r4sr)rZidscCs|jS)z7 Several types of numeric-dtypes Index objects )r)rr r r numeric_idx0s r Z5m4scCs t|jS)N)rr )r r r rrBscCs|jS)zT Several variants of Timedelta scalars representing 5 minutes and 4 seconds )r)rr r r scalar_td?sr!H)Zdaysz72:00:00DhcCs t|jS)N)rr )r r r rrPscCs|jS)za Several timedelta-like and DateOffset objects that each represent a 3-day timedelta )r)rr r r three_daysJs r&rx)Zhoursi )ZsecondsmcCs t|jS)N)rr )r r r rr_scCs|jS)zb Several timedelta-like and DateOffset objects that each represent a 2-hour timedelta )r)rr r r two_hoursYs r))ZminutesscCs|jS)zp Several timedelta-like and DateOffset instances that are _not_ compatible with Hourly frequencies. )r)rr r r not_hourlymsr,z23:00:00cCs|jS)zo Several timedelta-like and DateOffset instances that are _not_ compatible with Daily frequencies. )r)rr r r not_dailyxsr/imcCs|jS)z{ Several timedelta-like and DateOffset instances that are _not_ compatible with Monthly or Annual frequencies. )r)rr r rmismatched_freqsr0cCs|jS)z Several array-like containers that should have effectively identical behavior with respect to arithmetic operations. )r)rr r rboxsr1)ZmarkscCs|jS)zN Fixture equivalent to `box` fixture but xfailing the DataFrame case. )r)rr r r box_df_fails r2FTcCs|jS)z{ Fixture similar to `box` but testing both transpose cases for DataFrame, with the tranpose=True case xfailed. )r)rr r rbox_transpose_fails r3cCs|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes )r)rr r rbox_with_arraysr4)8ZnumpyrZpytestZ pandas.compatrZpandasZpdZpandas.util.testingutilZtestingZtmrZfixturerrrZIndexZzerosextendrrrZ Float64IndexZarangeZ Int64IndexZ UInt64IndexZ RangeIndexr Z TimedeltaZto_pytimedeltaZto_timedelta64r!ZoffsetsZDayZHourZ timedelta64r&ZMinuter)Z YearBeginZ MonthBeginZ_common_mismatchr,r/r0ZSeriesZ DataFramer1rZmarkZxfailr2r3Zto_arrayr4Zbox_with_array2r r r rs~  $