U >v f @sHddlZddlZddlZddlmZmZddlmZ ddl m Z ej dddgddgdd d Zej d ejd ejd gd ddZddeejejfDZeddeejfDeddejejejfDeejdejd gedddgej ed ddZej eejddd eejddd eejddd edgddddd Zej ed!ed!ed!gd"ddd#d$Zej ejd%ejd&ejd%d'ed(e d%d)e d&d*gd+ddd,d-Z!ej ejd.ej"d/ejd.d0ejd1d2e d.d*e d/d3gd4ddd5d6Z#ej$d.ej%d ej"gZ&ej ejd7d8e d7d9ejd7d2ge&d d:d;Z'ej e dge&d d?d@Z(ej e dAd)ejdAd'ejdAd'ge&d dBdCZ)ej eej*e j+eje,gdDdddEdFZ-dS)GN)Index RangeIndex) expressionsTi@BZnumexprpython)Zautouseparamsidsccs tj}|jt_|jV|t_dSN)expr _MIN_ELEMENTSparam)requestr r D/tmp/pip-unpacked-wheel-vdrwu74i/pandas/tests/arithmetic/conftest.pyswitch_numexpr_min_elements srdtype)rcCs|jS)a@ Several variants of integer value 1. The zero-dim integer array behaves like an integer. This fixture can be used to check that datetimelike indexes handle addition and subtraction of integers and zero-dimensional arrays of integers. Examples -------- dti = pd.date_range('2016-01-01', periods=2, freq='H') dti DatetimeIndex(['2016-01-01 00:00:00', '2016-01-01 01:00:00'], dtype='datetime64[ns]', freq='H') dti + one DatetimeIndex(['2016-01-01 01:00:00', '2016-01-01 02:00:00'], dtype='datetime64[ns]', freq='H') r r r r ronesrcCs4g|],}tjtjtjfD]}|dgd|dqqS)rr)npint64uint64float64).0box_clsrr r r 2srcCs g|]}|dgdtjdqS)rr)rr)rrr r rr7scCsg|]}tjd|dqS)rr)rarray)rrr r rr8srgcCs|jS)a Several types of scalar zeros and length 5 vectors of zeros. This fixture can be used to check that numeric-dtype indexes handle division by any zero numeric-dtype. Uses vector of length 5 for broadcasting with `numeric_idx` fixture, which creates numeric-dtype vectors also of length 5. Examples -------- arr = RangeIndex(5) arr / zeros Index([nan, inf, inf, inf, inf], dtype='float64') rrr r rzero@sr rrrrcCs t|jSrtype__name__xr r r`r&)rrcCs|jS)z7 Several types of numeric-dtypes Index objects rrr r r numeric_idxXsr(Z10m7scCs t|jSrr!r$r r rr&sr'cCs|jS)zV Several variants of Timedelta scalars representing 10 minutes and 7 seconds. rrr r r scalar_tdms r)H)daysz72:00:00DhcCs t|jSrr!r$r r rr&r'cCs|jS)za Several timedelta-like and DateOffset objects that each represent a 3-day timedelta rrr r r three_days|sr/x)hoursi )secondsmcCs t|jSrr!r$r r rr&r'cCs|jS)zb Several timedelta-like and DateOffset objects that each represent a 2-hour timedelta rrr r r two_hourssr5)minutesscCs|jS)zp Several timedelta-like and DateOffset instances that are _not_ compatible with Hourly frequencies. rrr r r not_hourlys r9z23:00:00cCs|jS)zo Several timedelta-like and DateOffset instances that are _not_ compatible with Daily frequencies. rrr r r not_dailys r<imcCs|jS)z{ Several timedelta-like and DateOffset instances that are _not_ compatible with Monthly or Annual frequencies. rrr r rmismatched_freqs r=cCs|jSr)r#r$r r rr&r'cCs|jS)zc Fixture to test behavior for Index, Series, tm.to_array, numpy Array and list classes rrr r r box_1d_arraysr>).ZnumpyrZpytestZpandaspdrrZpandas._testingZ_testingtmZpandas.core.computationrr Zfixturerrrrzerosextendrrr Zaranger(Z TimedeltaZto_pytimedeltaZto_timedelta64r)offsetsZDayZHourZ timedelta64r/ZMinuter5Z YearBeginZ MonthBeginZ_common_mismatchr9r<r=ZSeriesZto_arraylistr>r r r rs