o ;Dit @stddlZddlZddlmZddlmZmZmZm Z m Z ddZ ddZ dd Z d d Zd d ZddZddZdS)N)partial) randomseed randomtable RandomTable dummytable DummyTablecCs@t}tdt}t|tsJ|dksJ||ksJdS)zJ Ensure that randomseed provides a non-empty string that changes. N)rtimesleep isinstancestr)seed_1seed_2rS/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/test/util/test_random.pytest_randomseeds   rcCs>d\}}t||}t|d|ksJt||dksJdS)z] Ensure that randomtable provides a table with the right number of rows and columns. ) rrN)rlencolumnsrowstablerrrtest_randomtables rcCs@d\}}t||d}t|d|ksJt||dksJdS)z] Ensure that RandomTable provides a table with the right number of rows and columns. )<)numfldsnumrowsrrN)rrrrrrtest_randomtable_class s rcCsxdttjddfdttjgdfdttjgdfdtjff}d }t||d }|dd ks0Jt||d ks:Jd S)z Ensure that dummytable provides a table with the right number of rows and that it accepts and uses custom column names provided. countrdpet)dogcatcowcolor)yelloworangebrownvalue#)rfields)r r"r&r*rN)rpyrandomrandintchoicerandomrrrrrrtest_dummytable_custom_fields+s r1cCs:d}t|d}t|ddksJt||dksJdS)z~ Ensure that dummytable provides a table with the right number of rows and columns when not provided with a seed. r+rrrrNrrrrrrrtest_dummytable_no_seed=s r5cCs@d}d}t||d}t|ddksJt||dksJdS)z Ensure that dummytable provides a table with the right number of rows and columns when provided with an integer as a seed. r+*)rseedrrrNr3)rr7rrrrtest_dummytable_int_seedIs  r8cCs&d}t|d}t||dksJdS)z` Ensure that DummyTable provides a table with the right number of rows and columns. Fr2rN)rrr4rrrtest_dummytable_classUs r:)r0r-r functoolsrpetl.util.randomrrrrrrrrr1r5r8r:rrrrs