ó 2ÄÈ[c@`sÀdZddlmZmZmZddlZddlZddlm Z m Z m Z ddl m Z mZmZmZmZmZmZmZdefd„ƒYZdefd „ƒYZdS( s-Test functions for 1D array set operations. i(tdivisiontabsolute_importtprint_functionN(tassert_array_equalt assert_equalt assert_raises(tediff1dt intersect1dtsetxor1dtunion1dt setdiff1dtuniquetin1dtisint TestSetOpscB`sªeZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„Zd„ZRS(cC`stjddddgƒ}tjdddddgƒ}tjdddgƒ}t||dtƒ}t||ƒtjdddddgƒ}tjdddddddgƒ}tjdddgƒ}t||ƒ}t||ƒtgtggƒƒdS(Niiiiiit assume_unique(tnptarrayRtTrueR(tselftatbtectcted((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyttest_intersect1ds $ cC`svdtfd„ƒY}|ƒ}t||ƒ}t||ƒtdddgdddgƒ}t|dddgƒdS(NtTestcB`seZd„ZRS(cS`s tjdƒS(Ni(Rtarange(R((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyt __array__&s(t__name__t __module__R(((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyR%siii(tobjectRR(RRRtres((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyttest_intersect1d_array_like#s   !c C`stjddddgƒ}tjddddgƒ}t||dtdtƒ\}}}tjdddgƒ}t||ƒt|||ƒt|||ƒtjdddddddgƒ}tjddddddddgƒ}t||dtƒ\}}}tjddddgƒ}t||ƒt|||ƒt|||ƒtjddd dgd ddd ggƒ}tjddd dgd d ddggƒ}t||dtdtƒ\}}}tj||jƒ}tj||jƒ} tjddd dgƒ} t| ||ƒt| || ƒtjddd ddgdd dd dggƒ}tjddd d gd d dd ggƒ}t||dtƒ\}}}tj||jƒ}tj||jƒ} tjdd dgƒ} t| ||ƒt| || ƒdS(NiiiiiRtreturn_indicesiiiii i i (RRRRRt unravel_indextshape( RRRRti1ti2teeteftui1tui2tea((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyttest_intersect1d_indices/s<$ $' --$3-cC`sgtjddddgƒ}tjdddddgƒ}tjdddgƒ}t||ƒ}t||ƒtjdddgƒ}tjdddgƒ}tjddddddgƒ}t||ƒ}t||ƒtjddddgƒ}tjddddgƒ}tjddddddgƒ}t||ƒ}t||ƒtgtggƒƒdS( Niiiiiiii(RRRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyt test_setxor1dVs  ! ! c C`sÃtjgƒ}tjdgƒ}tjddgƒ}tgt|ƒƒtdgt|ddƒƒtdgt|ddƒƒtddgt|ddddƒƒtgt|ƒƒtdgt|ƒƒtdddgt|ddddƒƒtd d ddd gt|dd d gddd gƒƒtddgt|ddƒƒtddd gt|ddd gƒƒtddgt|ddƒƒtd d dgt|dd d gƒƒdS( Niiitto_begintto_endiÿÿÿÿii iii(RRRR(Rt zero_elemtone_elemttwo_elem((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyt test_ediff1dns%(:(c `sGd„}tj|dtgddhƒ‰‡fd†}tjdƒjddd gƒ}tjd d d gd ddgdddggƒ}|||ƒddg}d}|||ƒtjdƒ}|||ƒ|||ƒ|||ƒ|d|ƒ||dƒ|ddƒg}|||ƒ|||ƒ|||ƒdS(NcS`s%tj|ƒjƒjƒ}||kS(N(Rtasarraytflattenttolist(RR((s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyt _isin_slow„stotypestexcludedic`s/t||ƒ}ˆ||ƒ}t||ƒdS(N(R R(RRtxty(t isin_slow(s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pytassert_isin_equalˆsiiiii iiii ii!i iiii(i i(ii(i i(Rt vectorizetboolRtreshapeR( RR7R=RRRtdtfR:((R<s?/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_arraysetops.pyt test_isin€s( !!3           cC`sÈxYd D]Q}ddddg}dddddg|}tjttttgƒ}t||dtƒ}t||ƒd |d s  :ÿH