o !wi@s~ddlZddlZddlmZmZddlmZmZmZmZm Z ddl m Z eddZ e ddZ d gZ d d iZe d d ZdS)N)BuiltinFunctionType FunctionType)curryidentity keyfilter valfilter merge_with) dev_skip_testcCs*t|ddpd}||pdtt|vS)N __module__ztoolz.functoolz.curry)getattr startswithstrtype)modnamefuncmodr]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/cytoolz/tests/test_embedded_sigs.py isfrommod src s:ddl}ttd|jttdtjtddtfddtfddGd d d tt}|D]\\}\}}t |t rOt |}nt ||j r\t |j}nt |j}|jfd d |jDd }|j}|dddd}|jt|}||vs||vsd|||jf} J| q>dS)z Test that all ``cdef class`` extension types in ``cytoolz`` have correctly embedded the function signature as done in ``toolz``. rNtoolzcytoolzcSs t|t SN) isinstancerxrrrs z!test_class_sigs..c|vSrrr) cytoolz_dictrrrcrrrr) toolz_dictrrrrc@s eZdZdZddZddZdS)ztest_class_sigs..wrapzNe.g., allow `factory=` to instead be `factory=dict` in signaturecSs ||_dSr)obj)selfr!rrr__init__#s z&test_class_sigs..wrap.__init__cSst|jdt|jS)N__name__)r r!repr)r"rrr__repr__&sz&test_class_sigs..wrap.__repr__N)r$r __qualname____doc__r#r&rrrrwrap!s r)cs.g|]}|jtjur|j|jdn|qS))default)r*inspect_emptyreplace).0v)r)rr 5s  z#test_class_sigs..) parametersz Py_ssize_t r z=u'z='z_cytoolz.%s does not have correct function signature. Expected: %s Docstring in cytoolz is: %s)rrr__dict__rrrritemsrrr+ signaturerrr#r-r1valuesr(r$r) rdkey toolz_func cytoolz_func toolz_specdocdoc_alt toolz_sigmessager)rr r)rtest_class_sigss>        r?rcompcomposecCsttdtj}tdd|}|D]I\}}|j}t|dks+Jd||jfdt ||}||dvsBJd||jf|d rOJd ||jf|ds\Jd ||jfqd S) z~ Test that the function signature is at the beginning of the docstring and is followed by exactly one blank line. rcSs|tvSr) skip_sigsrrrrrRrz'test_sig_at_beginning..z#cytoolz.%s docstring too short: %sz%s(rz8cytoolz.%s docstring missing signature at beginning: %szcytoolz.%s docstring too many blank lines after signature: %sN) rrrr2rr3r( splitlineslenaliasesget)rr7valdoclinessigrrrtest_sig_at_beginningLs0    rL)r+rtypesrrrrrrrr rr?rBrGrLrrrrs   8