DUfGlddlZddlZddlZddgZdddddiZd ZGd dZdd Z dd Z dZ dS)Nupdate_default_colnamesis_chrom_dtypecolnameschromstartendrrrcrtddtddtddfS)z Returns default column names. These defaults be updated with :func:`update_default_colnames`. Returns ------- colnames : triplet (str, str, str) rrrr)_rcP/var/www/html/software/conda/lib/python3.11/site-packages/bioframe/core/specs.py_get_default_colnamesrs, z?7 #S_W%=s:u?U UUr c eZdZdZdZdZdS)rcttd|_t|tjjrTt|dkrtd|\tdd<tdd<tdd<dSt|tjj r>td d| DdStd)Nrz@Please, specify new columns using a list of 3 strings or a dict!rrrc"i|] \}}|dv || S)r r ).0kvs r z4update_default_colnames.__init__..,s41555q555r ) dictr _old_colnames isinstance collectionsabcIterablelen ValueErrorMappingupdateitems)self new_colnamess r__init__z update_default_colnames.__init__s!#j/22 lKO$< = = <  A%% +  J(J(J&&&  ko&= > >   O " " , 2 2 4 4     U r c|S)Nr )r#s r __enter__z!update_default_colnames.__enter__7s r c$|jtd<dS)Nr)rr )r#argss r__exit__z update_default_colnames.__exit__:s,Jr N)__name__ __module__ __qualname__r%r'r*r r rrrsA4-----r Fc t|tjs|rdStd|rs b", ' '2  50111< s8}}  H - -:;; ; x== ! !"* - -  5 IIc(mm..s2:?? @ @* +   tr c|tn|\}}}t||||gds|rdStd|j|||g\}}}t |s|rdSt dt jj |s|rdSt dt jj |s|rdSt d|rdSdS) a Checks that dataframe `df` has chrom, start, end columns with valid dtypes. Raises TypeErrors if cols have invalid dtypes. Parameters ---------- df : pandas.DataFrame cols : (str, str, str) or None The names of columns containing the chromosome, start and end of the genomic intervals, provided separately for each set. The default values are 'chrom', 'start', 'end'. return_as_bool : bool If true, returns as a boolean instead of raising errors. Default False. NT)r8Fzcould not verify columnszAinvalid df['chrom'] dtype, must be object, string, or categoricalz*invalid df['start'] dtype, must be integerz(invalid df['end'] dtype, must be integer) rr9rdtypesr TypeErrorr/apitypesis_integer_dtype) r6colsr8ck1sk1ek1 chrom_dtype start_dtype end_dtypes r_verify_column_dtypesrGas*$04|)+++MCc 2S#t D D D5  53444*,)S#sO*D'Ki + & &  5 O    6< ( ( 5 5F  5DEEE 6< ( ( 3 3D  5BCCCtr ctjtjj|tjj|t|tjjjgS)z] Returns True if dtype is any of the allowed bioframe chrom dtypes, False otherwise. ) npanyr/r=r>is_string_dtypeis_object_dtyperCategoricalDtype)rDs rrrs[ 6 FL ( ( 5 5 FL ( ( 5 5 {BFL$A B B   r )FF)NF) rnumpyrIpandasr/__all__r rrr9rGrr r rrQs  WwuEEF V V V--------D    F++++\     r