a < bd @sddlmZddlZddlZddlZddlmZmZmZm Z m Z ddZ ddZ dd Zd d Zd d ZddZddZddZddZddZddZddZddZejdddgd d!Zd"d#Zd$d%Zd&d'Zd(d)Zejd*d+d,d-gfd.dd-gfgd/d0Z d1d2Z!d3d4Z"ejd5gd6fgd7fgd8fgd9fgd:d6fgd:d9fgd:d8fgd:d9fgd;d<Z#d=d>Z$d?d@Z%dAdBZ&dCdDZ'dEdFZ(dS)G)datetimeN) DataFrameIndex MultiIndexSeries_testingcCsVtdtjdg|d}tjtdd |jjdddWdn1sH0YdS)NfooBAD__barBADfoodtypezexpand must be True or Falsematch.*(BAD[_]+).*(BAD)expand)rnpnanpytestraises ValueErrorstrextract)any_string_dtypevaluesrq/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/strings/test_extract.py+test_extract_expand_kwarg_wrong_type_raisessrcCstdtjdg|d}tdtjtjg|d}|jd}t|||jjddd}t||tddgtjtjgtjtjgg|d}|jjd d d}t||dS) Nrr r BAD__z .*(BAD[_]+).*TrBADrF)rrrrrrtmassert_frame_equal)rsexpectedresultrrrtest_extract_expand_kwargs   r$c Cstdtjddtddddg }|jjddd }tjtjg}td d g|d d g||||||g }t |||jjd dd }td tjd tjtjtjtjtjtjg }t ||dS) NaBAD_BAD BAD_b_BADTr @rFrBAD_rz.*(BAD[_]+).*BAD) rrrrtodayrrrrr assert_series_equal)Zserr#err"rrr&test_extract_expand_False_mixed_object(s " "r-cCsRtgd}d}tjt|d |jjdddWdn1sD0YdS)N)A1A2A3ZA4ZB5z,only one regex group is supported with Indexr ([AB])([123])Fr)rrrrrr)idxmsgrrr test_extract_expand_index_raises;s r4cCs|gd|d}d}tjt|d |jjdddWdn1sH0Ytjt|d |jjdddWdn1s0YdS) Nr.B2ZC3r "pattern contains no capture groupsr [ABC][123]Fr (?:[AB]).*rrrrrindex_or_seriesrs_or_idxr3rrr,test_extract_expand_no_capture_groups_raisesEs .r>cCsX|ddg|d}|jjddd}|ddgd|d }|tkrHt||n t||dS) Nr.r/r (?PA)\dFrAunonamer )rrrrr+Zassert_index_equalr<rr=r#r"rrr(test_extract_expand_single_capture_groupRs rEcCstgd|d}|jjddd}ttjtjtjg|d}t|||jjddd}ttjtjgtjtjgtjtjgg|d}t|||jjddd}tdd tjg|d}t|||jjd dd}tdd gd d gtjtjgg|d}t|||jjd dd}tdd tjgd|d}t|||jjddd}tdd gd d gtjtjggddg|d}t|||jjddd}tdd gd d gtjtjggddg|d}t|||jjddd}tdd tjg|d}t||tgd|d}|jjddd}tdd gd d gtjtjgg|d}t||tgd|d}|jjddd}tdd gd d gtjdggddg|d}t||tgd|d}|jjddd}tdd gd d gdtjggddg|d}t||dS)Nr5r (_)Fr(_)(_) ([AB])[123]r@Br112(?P[AB])letterrB!(?P[AB])(?P[123])numbercolumnsr ([AB])(?P[123])r([AB])(?:[123])ZA11ZB22ZC33([AB])([123])(?:[123])r.r63"(?P[AB])?(?P[123])rWr.r6C#(?P[ABC])(?P[123])?rZ) rrrrrrr+rr rr!r#r"rrr"test_extract_expand_capture_groups^sv "         r]cCsgd}t|t|kr"td|dt|}t|||d}|jjddd}tddtjg||d}t |||jjd dd}t d dgd dgd tjggd dg||d}t ||dS)NrYIndex too shortindexr (\d)FrrJrK(?P\D)(?P\d)?r@rIrZrMrOrQr`r ) lenrskiprrrrrrr+rr r`rdatar!r#r"rrr(test_extract_expand_capture_groups_indexs   rhcCsDtgdd|d}|jjddd}tgdd|d}t||dS) Na3b3c2ZbobrBz(?P[a-z])FrabcZsue)rrrrr+r\rrr,test_extract_single_series_name_is_preservedsrqcCsZtdtjdg|d}|jjddd}tddgtjtjgtjtjgg|d}t||dS) Nrr r rTrrr)rrrrrrrr r\rrrtest_extract_expand_Trues rrc Csntjtjg}tdtjddtddddg }|jjddd}td d g|d d g||||||g }t ||dS) Nr%r&Tr r'r(rrr)r) rrrrr*rrrrr )r,mixedr#r"rrr%test_extract_expand_True_mixed_objects  "rtcCs|gd|d}d}tjt|d |jjdddWdn1sH0Ytjt|d |jjdddWdn1s0YdS) Nr5r r7r r8Trr9r:r;rrr4test_extract_expand_True_single_capture_group_raisess .rucCsD|ddg|d}|jjddd}tdddgi|d}t||dS) Nr.r/r r?TrrAr@)rrrrr rDrrr-test_extract_expand_True_single_capture_groupsrvrC series_namecCstgd||d}|jjddd}ttjtjtjg|d}t|||jjddd}ttjtjgtjtjgtjtjgg|d}t|||jjddd}td d tjg|d}t|||jjd dd}td d gd d gtjtjgg|d}t|||jjddd}tdd d tjgi|d}t|||jjddd}td d gd d gtjtjggddg|d}t|||jjddd}td d gd d gtjtjggddg|d}t|||jjddd}td d tjg|d}t||dS)Nr5rBrFTrr rGrHr@rIr1rJrKrLrMrNrOrPrRrrSrrrrrrrr )rCrr!r#r"rrrtest_extract_seriessJ "      rycCstgd|d}|jjddd}tddgdd gtjtjgg|d}t||tgd |d}|jjd dd}tddgdd gtjd ggd dg|d}t||tgd|d}|jjddd}tddgdd gdtjggd dg|d}t||dS)NrTr rUTrr@rJrIrKrVrXrWrMrOrPrYr[rZrxr\rrrtest_extract_optional_groupsBs,  rzcCsgd}t|t|kr"td|dt|}t|||d}|jjddd}tddtjg||d}t |||jjd dd}td dgd dgd tjggd dg||d}t ||dS)NrYr^r_raTrrJrKrbr@rIrZrMrOrc) rdrrerrrrrrrr rfrrr+test_extract_dataframe_capture_groups_indexas   r{cCsFtgdd|d}|jjddd}tdgdi|d }t||dS) NrirwrB(?P[a-z])TrrMrmr )rrrrrr r\rrr'test_extract_single_group_returns_frame{sr}c Csdddddtjdg}gd}d}gd }t||d }tjgd d d }t||||d }|jj|tj d}t ||tgd} t|| |d}tjgddd }t||||d }|jj|tj d}t ||t|| |d}d|j _ d|_ t||||d }|jj|tj d}t ||dS)Nzdave@google.comztdhock5@gmail.comzmaudelaperriere@gmail.comz'rob@gmail.com some text steve@gmail.comz%a@b.com some text c@d.com and e@f.com))ZdaveZgooglecom)Ztdhock5gmailr)Zmaudelaperriererr)Zrobrr)Zsteverr)rnror)rpdr)efrzY (?P[a-z0-9]+) @ (?P[a-z]+) \. (?P[a-z]{2,4}) )userdomaintldr )rrr'rr)r)rr')r)rr')rrNr names)flags))singleDave)rToby)rMaude)multiple robAndSteve)rabcdef)nonemissing)remptyr_))rrr)rrr)rrr)rrr)rrr')rrr)rrr')rrr)NNr )matches description)rrr )rrrr from_tuplesrr extractallreVERBOSErr r`r) rrgZexpected_tuplespatZexpected_columnsr!expected_indexr"r#mirrrtest_extractallsT        rzpat,expected_namesrXrMrOz([AB])?(?P[123])cCs\tgd|d}|j|}tdtjdftjdfgtjgddd||d }t ||dS) N)r~r.Z32r )r@rJrWrK)rrrr'rr)r`rQr ) rrrrrrrrrr )rZexpected_namesrr!r#r"rrrtest_extractall_column_namess  rcCstgdd|d}tjgddd}|jd}tdgd i||d }t|||jd }tgd ||d }t||dS) NrjrkZd4c2rwrBrrrrrrr|rM)rnrorrpr_([a-z]))rrrrrrrr )rr!rr#r"rrrtest_extractall_single_groups    rcCsNtgdd|d}|jd}tgdtjgddd|d }t||dS) N)Zab3Zabc3Zd4cd2rwrBz([a-z]+))ababcrZcdrrrr_)rrrrrrrr r\rrr,test_extractall_single_group_with_quantifiers rz data, names)N)i1)Ni2)rrrc sBt|t|dkr*tt|dd}n$fddtD}tj||d}t|d||d}tjg|d d}|jd }tdg||d }t |||jd }tddg||d }t |||jd }tdg||d }t |||jd}tddg||d }t |||jd}tddg||d }t ||dS)Nr'rrCc3s |]}t|gdVqdS)r'N)tuple).0inrr (z-test_extractall_no_matches..rrwrCr`r r z(z)rcz(z)(z)z (?Pz)firstz(?Pz)(?Pz)secondz(z)(?Pz)) rdrrangerrrrrrrr ) rgrrr`Ztuplesr!rr#r"rrrtest_extractall_no_matchess8            rcCstgdd|d}|jd}tdgditjgdddgd |d }t|||d krtgdtgddd fD]}|jd}t||qvtgdd tgddd |d}|jd}tdgditjgdddgd |d }t||dS)N)Za1a2b1Zc1ZxxxrBz[ab](?P\d)digit)rJrKrJ)r)rr'rr rr_objectrZs_name)XXyyzzZidx_namer))rr)rr')rr) rrrrrrrr r)rr!r#r"r2rrrtest_extractall_stringindexMs:        rcCsPtgdd|d}tjtdd|jdWdn1sB0YdS)NrrwrBzno capture groupsr z[a-z])rrrrrr)rr!rrr(test_extractall_no_capture_groups_raisesssrcCsztgdgddd}|jjjddd}tgd}t|||jjjd dd}gd }t|d d gd }t||dS)Nr)r0ZB3ZD4rw)r`rCz([A-Z])Tr)r@rIDz!(?P[A-Z])(?P[0-9])))r@rW)rIrW)r4rMr)rQ)rr`rrrrr )r!rrZe_listrrr!test_extract_index_one_two_groups{s  rc Cstgdd|d}d}|jj|dd}|j|}|jddd }t||d }|jj|dd}|j|}|jddd }t||d }|jj|dd} |j|}|jddd }t| |d } |jj| dd} |j| }|jddd }t| |dS) NrirwrB([a-z])([0-9])Trrr level!(?P[a-z])(?P[0-9])(?P[a-z])r)rrrrxsrr ) rr!pattern_two_nonameextract_two_nonameZhas_multi_indexZno_multi_indexpattern_two_namedextract_two_namedpattern_one_namedextract_one_namedpattern_one_nonameextract_one_nonamerrrtest_extractall_same_as_extracts*       rc Cstjgddd}tgd|d|d}d}|jj|dd }|j|}|jd d d }t||d }|jj|dd }|j|}|jd d d }t||d} |jj| dd } |j| }|jd d d }t| |d} |jj| dd } |j| }|jd d d }t| |dS)N))r@r)rIr)rZthird)ZcapitalZordinalrrirw)r`rCr rTrrr rrrr) rrrrrrrrr ) rrr!rrZhas_match_indexZno_match_indexrrrrrrrrr-test_extractall_same_as_extract_subject_indexs2       r))rrZnumpyrrZpandasrrrrrrrr$r-r4r>rEr]rhrqrrrtrurvmarkZ parametrizeryrzr{r}rrrrrrrrrrrrrrsd     S  9 U        +&