o qo™i|<ã@sôdZddlZddlZddlZddlZddlZddlZddlmZm Z ddl mZddl mZddl mZejGdd„dƒƒZejGd d „d ƒƒZddd„Zd d„Zdd„ZGdd„dƒZGdd„dejƒZGdd„dejƒZedkrxe ¡dSdS)z#Tests for partial_with_arg_factory.éN)ÚListÚAny)Úabsltest)Ú parameterized)Úarg_factoryc@seZdZUdZeed<dS)ÚChildrÚxN)Ú__name__Ú __module__Ú__qualname__rÚintÚ__annotations__©rrúP/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/arg_factory_test.pyrs rc@sReZdZUejedZeee d<dd„Z e e e¡Z eje e¡fdd„ƒZdS)ÚParent)Údefault_factoryÚchildrencCs|j |¡dS©N)rÚappend©ÚselfÚchildrrrÚ add_child'szParent.add_childcCs| |¡dSr)rrrrrÚnew_child_2,szParent.new_child_2N)r r rÚdataclassesÚfieldÚlistrrrr rrÚ partialmethodÚ new_childÚsupply_defaultsrrrrrrr#s rcCó||||fSrr©Úarg1Úarg2Úkwarg1Úkwarg2rrrÚ sample_fn1ór&cCó |||fSrr©ÚaÚbÚcrrrÚ posonly_fn5ó r-cOó||fSrr)ÚargsÚkwargsrrrÚ varargs_fn9ór2c@seZdZddd„Zdd„ZdS)ÚCounterÚcountercCs||_d|_dS)Nr)ÚnameÚcount)rr6rrrÚ__init__?s zCounter.__init__cCs |jd7_|j›d|j›S)NéÚ_)r7r6©rrrrÚ__call__CszCounter.__call__N)r5)r r rr8r<rrrrr4=s r4c@sÚeZdZdd„Zdd„Zdd„Ze d¡dd „ƒZd d„Z dd „Z dd„Zdd„Zdd„Z e d¡dd„ƒZdd„Zdd„Zdd„Ze ejeeddfejeeddfg¡d d!„ƒZd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*S)+ÚArgFactoryPartialTestcCsXt ttƒ¡}| |dƒd¡| |dƒd¡| |dddd¡| |ddd ¡dS) Né)Ú counter_1r>rN)Ú counter_2r>rNéé©r%)Ú counter_3rArrB©r#)Ú counter_4r>rN©rÚpartialr&r4ÚassertEqual©rÚprrrÚtest_factory_positional_argJs z1ArgFactoryPartialTest.test_factory_positional_argcCsNtjttƒd}| |ddƒd¡| |ddƒd¡| |ddddd¡dS) N©r$r9rA©r9rAr?N©r9rAr@NrrC©r9rArDrrGrJrrrÚtest_factory_keyword_argQsz.ArgFactoryPartialTest.test_factory_keyword_argcCsLtjttƒd}| |ddd¡| |dddd¡| |ddd ¡dS) NrEr>©r"©r>r?rNr9rA©r"r#©r9rArN©r>r@rNrGrJrrrÚ'test_factory_is_not_run_when_overriddenWsz=ArgFactoryPartialTest.test_factory_is_not_run_when_overriddenz,Enable this after dropping pyhon 3.7 supportcCs:dd„}tj|dd„dd„d}| |ƒddd if¡dS) Nc[r/rr)rr1rrrÚfcóz:ArgFactoryPartialTest.test_positional_only_args..fcSódS)NÚpositional_xrrrrrÚiózAArgFactoryPartialTest.test_positional_only_args..cSrZ)NÚ keyword_xrrrrrr\ir]©rr[rr^©rrHrI©rrXrKrrrÚtest_positional_only_args`sz/ArgFactoryPartialTest.test_positional_only_argscCs|tjttdƒtdƒtdƒd}| |ƒd¡| |ddd¡| |dd d d¡| |dd dd dd¡| |ƒd¡dS)Nr"r#r%)r"r#r%©Úarg1_1Úarg2_1rÚkwarg2_1r>rR©r>Úarg2_2rÚkwarg2_2érT©r>rjrÚkwarg2_3éér!©r>rjrmrn©Úarg1_2Úarg2_3rÚkwarg2_4rGrJrrrÚtest_multiple_factorieslsüz-ArgFactoryPartialTest.test_multiple_factoriescCs6tƒ}tjt||d}| |ƒd¡| |ƒd¡dS)NrT©r?r@rN©rDrFrN)r4rrHr&rI©rÚfactoryrKrrrÚtest_reused_factoryxsz)ArgFactoryPartialTest.test_reused_factorycCsfdd„}tj|td}|dƒ}|dƒ}| |tdgd¡| |tdgd¡| |d|d¡dS)NcSó ||dœS©N©r6rrr|rrrrX€r.z2ArgFactoryPartialTest.test_list_factory..f©rÚjoeÚsuer|r)rrHrrIÚdictÚassertIsNot)rrXrKr~rrrrÚtest_list_factory~sz'ArgFactoryPartialTest.test_list_factorycCsHdd„}t |t ¡j¡}| |ƒd¡| |ƒd¡| |ƒd¡dS)NcSó|dS©NrArr_rrrr\ŒózDArgFactoryPartialTest.test_itertools_count_factory..rrAé)rrHÚ itertoolsr7Ú__next__rI)rÚdoublerKrrrÚtest_itertools_count_factory‹s z2ArgFactoryPartialTest.test_itertools_count_factorycCstt tdd„¡}|ƒ}|ƒ}| |ttdƒgƒ¡| |ttdƒgƒ¡| |j|j¡| |jd|jd¡dS)NcSó tdƒgS©NrB©rrrrrr\“ó zQArgFactoryPartialTest.test_customized_dataclass_default_factory..rBr)rrHrrIrrr©rÚparent_factoryr*r+rrrÚ)test_customized_dataclass_default_factory’sz?ArgFactoryPartialTest.test_customized_dataclass_default_factorycCs0dd„}tj|dd„d}| |ddd¡dS) NcSs||Srr)rÚotherrrrrX r3z6ArgFactoryPartialTest.test_param_named_self..fcSrZ)Né rrrrrr\£r]z=ArgFactoryPartialTest.test_param_named_self..)r’r>r;ér`rarrrÚtest_param_named_self›sz+ArgFactoryPartialTest.test_param_named_selfcCs2dd„}tj|tttd}| |ƒgggf¡dS)NcSr(rr©Úfuncr0r1rrrrXªr.z6ArgFactoryPartialTest.test_param_named_func..fr–)rrHrrIrarrrÚtest_param_named_func¦sz+ArgFactoryPartialTest.test_param_named_funccCóNtƒ}| ¡| ¡| |ttdƒtdƒgƒ¡| |jd|jd¡dS©Nrr9)rrrIrrr©rÚparentrrrÚtest_partialmethod°ó z(ArgFactoryPartialTest.test_partialmethodcCsÌ| t tjttd¡¡| t tjdd„td¡¡| t t tdƒj¡¡¡| t t jtdd¡¡| t t jdd„dd¡¡| t t tdƒj¡¡¡| t d¡¡| t d¡¡dS)Nr_cSó|dS©Nr9rr_rrrr\½r…zAArgFactoryPartialTest.test_is_a_factory_partial..rcSrŸr rr_rrrr\Çr…Úxyz) Ú assertTruerÚis_arg_factory_partialrHrrrÚ__eq__ÚassertFalseÚ functoolsr;rrrÚtest_is_a_factory_partial·s8ÿÿÿÿÿÿÿÿÿÿz/ArgFactoryPartialTest.test_is_a_factory_partialrRz9(*, arg1=, arg2, kwarg1=0, kwarg2=None)rMz7(arg1, arg2, *, kwarg1=, kwarg2=None)cCs| tt |¡ƒ|¡dSr)rIÚstrÚinspectÚ signature)rrKrªrrrÚtest_partial_func_signatureÎsz1ArgFactoryPartialTest.test_partial_func_signaturecCs:tjtdd}tj|td}| |dddgddf¡dS)Nr>rRrEr“rM)r¦rHr&rrrI©rÚp1Úp2rrrÚ/test_arg_factory_partial_wrap_functools_partial×ózEArgFactoryPartialTest.test_arg_factory_partial_wrap_functools_partialcCs:tjttd}tj|dd}| |dddgddf¡dS)NrEr>rRr“rM)rrHr&rr¦rIr¬rrrÚ/test_functools_partial_wrap_arg_factory_partialÜr°zEArgFactoryPartialTest.test_functools_partial_wrap_arg_factory_partialcCsx| td¡t td¡Wdƒn1swY| td¡tjtddWdƒdS1s5wYdS)NzExpected .* to be callabler>rM)ÚassertRaisesRegexÚ TypeErrorrrHr&r;rrrÚ)test_arg_factories_must_be_callable_errorásÿ"ÿz?ArgFactoryPartialTest.test_arg_factories_must_be_callable_errorcCsJtjttd}| td¡|ddƒWdƒdS1swYdS)N)r,z>got some positional-only arguments passed as keyword argumentsr9rA)rrHr-rr²r³rJrrrÚ9test_partial_cant_bind_positional_only_params_as_keywordsçsþ"ýzOArgFactoryPartialTest.test_partial_cant_bind_positional_only_params_as_keywordsN)r r rrLrQrWrÚskiprbrtryr‚rŠr‘r•r˜rr§rÚ parametersrrHr&rr«r¯r±r´rµrrrrr=Hs: ÿÿý r=c@s¤eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„Zdd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'S)(ÚDefaultFactoryTestcCs€tjt tƒ¡ddfdd„ƒ}| |ddd¡| |ddd¡| |dd d d¡| |ddd d¡| |ddd¡dS)NrcSr rrr!rrrrKóóz.DefaultFactoryTest.test_factory_arg..pr>rRrSrVrArB)r"r%)rArDrrBr9rTrU)r>rFrN©rrrr4rIrJrrrÚtest_factory_argñsÿz#DefaultFactoryTest.test_factory_argcCsZtjt tƒ¡dfdd„ƒ}| |ddƒd¡| |ddƒd¡| |ddddd ¡dS) NcSr rrr!rrrrKr¹z0DefaultFactoryTest.test_factory_kwarg..pr9rArNrOrrCrPrºrJrrrÚtest_factory_kwargÿsÿz%DefaultFactoryTest.test_factory_kwargcCsZtjt tƒ¡ddfdd„ƒ}| |ddd¡| |ddd d ¡| |ddd¡dS)NrcSr rrr!rrrrK r¹zEDefaultFactoryTest.test_factory_is_not_run_when_overridden..pr>rRrSr9rArTrUrVrºrJrrrrWsÿz:DefaultFactoryTest.test_factory_is_not_run_when_overriddencCs”tjt tdƒ¡t tdƒ¡dt tdƒ¡fdd„ƒ}| |ƒd¡| |dd d ¡| |dddd ¡| |dddddd¡| |ƒd¡dS)Nr"r#rr%cSr rrr!rrrrKóz5DefaultFactoryTest.test_multiple_factories..prcr>rRrgrjrTrkrmrnr!rorprºrJrrrrtsüz*DefaultFactoryTest.test_multiple_factoriescCsJtƒ}tjt |¡t |¡ddfdd„ƒ}| |ƒd¡| |ƒd¡dS)NrcSr rrr!rrrrK.r½z1DefaultFactoryTest.test_reused_factory..prurv)r4rrrrIrwrrrry+süz&DefaultFactoryTest.test_reused_factorycCshtjt t¡fdd„ƒ}|dƒ}|dƒ}| |tdgd¡| |tdgd¡| |d|d¡dS)NcSrzr{rr|rrrrX<ó z/DefaultFactoryTest.test_list_factory..fr~rr|r)rrrrrIr€r)rrXr~rrrrr‚:sz$DefaultFactoryTest.test_list_factorycCsLtjt t ¡j¡fdd„ƒ}| |ƒd¡| |ƒd¡| |ƒd¡dS)NcSrƒr„rr_rrrrKIrYz:DefaultFactoryTest.test_itertools_count_factory..prrAr†)rrrr‡r7rˆrIrJrrrrŠGs z/DefaultFactoryTest.test_itertools_count_factorycCs€tjt dd„¡fdd„ƒ}|ƒ}|ƒ}| |ttdƒgƒ¡| |ttdƒgƒ¡| |j|j¡| |jd|jd¡dS)NcSr‹rŒrrrrrr\UrŽzNDefaultFactoryTest.test_customized_dataclass_default_factory..cSst|ƒSr)rr}rrrrSszTDefaultFactoryTest.test_customized_dataclass_default_factory..parent_factoryrBr)rrrrIrrrrrrrrr‘Qsÿz.pr9rA)rrrrrIr)rrKrÚyrrrÚtest_posonlygs zDefaultFactoryTest.test_posonlycCsÆtjt t¡fdttfdd„ƒ}|dƒ}|dƒ}|d|ƒ}| |ddg¡| |dg¡| ||¡| ||¡| |j d¡| |j d¡| t|d ƒ¡| |j |jj ¡| |j|jj¡dS) NÚseqcSó| |¡|S)ú Docstring.©r)ÚeltrÁrrrrts z7DefaultFactoryTest.test_supply_defaults..appendrBr†r>rÃrÚ__wrapped__)rrrrrrrIÚassertIsrÚ__doc__r r¢ÚhasattrrrÆr )rrrr¿ÚzrrrÚtest_supply_defaultsss z'DefaultFactoryTest.test_supply_defaultscCsFtjt t¡dœdd„ƒ}| |dƒdgf¡| |dddd¡dS)N)r¿cSr/rr©rr¿rrrrXŠrYzADefaultFactoryTest.test_default_for_keyword_only_param..fr9rA©r9rA©rrrrrI©rrXrrrÚ#test_default_for_keyword_only_paramˆsz6DefaultFactoryTest.test_default_for_keyword_only_paramcCsBtjt t¡fdd„ƒ}| |dƒdgf¡| |ddƒd¡dS)NcSr/rrrÌrrrrX“rYzDDefaultFactoryTest.test_default_for_positional_only_param..fr9rArÍrÎrÏrrrÚ&test_default_for_positional_only_param‘sz9DefaultFactoryTest.test_default_for_positional_only_paramcCs@| td¡t dd„¡WdƒdS1swYdS)Nz&expected at least one argument to havecSr/rrrÌrrrr\žr…zQDefaultFactoryTest.test_supply_defaults_with_no_default_factory..)r²Ú ValueErrorrrr;rrrÚ,test_supply_defaults_with_no_default_factoryšs ÿ"ýz?DefaultFactoryTest.test_supply_defaults_with_no_default_factorycCs˜d}t t¡fdd„}t tj¡fdd„}| t|¡|dƒWdƒn1s*wY| t|¡ |dƒWdƒdS1sEwYdS)Nz‚arg_factory\.default_factory$.*$ does not support.*\n.*\nDid you forget to apply the `@arg_factory\.supply_defaults` decorator\?cSrÂrrÄ)rÚlstrrrrX¦s z:DefaultFactoryTest.test_missing_decorator_error..fcSs|||Srr)rÚnoiserrrÚgªr'z:DefaultFactoryTest.test_missing_decorator_error..gr>)rrrÚrandomr²ÚAttributeErrorrÒ)rÚexpected_errorrXrÖrrrÚtest_missing_decorator_error sÿ ÿ "ÿz/DefaultFactoryTest.test_missing_decorator_errorcCó4tjt t¡fdttfdd„ƒ}| |ƒg¡dS)NrcSó|Srrr_rrrrX´óz@DefaultFactoryTest.test_default_factory_type_override..f)rrrrrrrIrÏrrrÚ"test_default_factory_type_override²óz5DefaultFactoryTest.test_default_factory_type_overridecCrÛ)NrcSrÜrrr_rrrrX¼rÝz.f)rrÚ ArgFactoryrrrrIrÏrrrÚtest_arg_factory_type_overrideºrßz1DefaultFactoryTest.test_arg_factory_type_overridecCs6t t¡}t t¡}| |t t¡¡| ||¡dSr)rràrr€rIÚassertNotEqual)rÚlist_factoryÚdict_factoryrrrÚtest_arg_factory_equalÂs z)DefaultFactoryTest.test_arg_factory_equalcCs2t t¡}| |t |¡¡| |t |¡¡dSr)rràrrIÚcopyÚdeepcopy)rrxrrrÚ"test_arg_factory_copy_and_deepcopyÈs z5DefaultFactoryTest.test_arg_factory_copy_and_deepcopyN)r r rr»r¼rWrtryr‚rŠr‘rrÀrËrÐrÑrÓrÚrÞrárårèrrrrr¸ïs( r¸Ú__main__)rN)rÈrærr¦r©r‡r×ÚtypingrrÚabsl.testingrrÚfiddlerÚ dataclassrrr&r-r2r4ÚTestCaser=r¸r ÚmainrrrrÚs4 (`ÿ