o j @svdZddlZddlZddlmZddlmZGdddeZdddZ ifd d Z e d kr9d d Z ej d ddSdS)zgSelf tests These tests should perform quickly and can ideally be used every time an application runs. N) import_module)StringIOc@seZdZddZdS) SelfTestErrorcCst|||||_||_dSN) Exception__init__messageresult)selfrr r /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/SelfTest/__init__.pyr&s zSelfTestError.__init__N)__name__ __module__ __qualname__rr r r r r%s rc Ks|duri}t}|dur|durt|d}||n|dur+||j|dntd|dur9t|d<n||d<tjdd|i|}||}|sc|dur^t j |d t d||S)a>Execute self-tests. This raises SelfTestError if any test is unsuccessful. You may optionally pass in a sub-module of SelfTest if you only want to perform some of the tests. For example, the following would test only the hash modules: Cryptodome.SelfTest.run(Cryptodome.SelfTest.Hash) Nconfigz5'module' and 'tests' arguments are mutually exclusivestream verbosityzSelf-test failedr )unittest TestSuite get_testsaddTests ValueErrorrTextTestRunnerrun wasSuccessfulsysstderrwritegetvaluer) modulerrtestsrkwargssuiterunnerr r r r r,s(      rcCs6g}gd}|D]}td|}||j|d7}q|S)N) CipherHashProtocol PublicKeyRandomUtil SignatureIOMathzCryptodome.SelfTest.r)rr)rr! module_namesnamer r r r rRs  r__main__cCs ttSr)rrrr r r r r#bs r#) defaultTest)NrNNN)__doc__rr importlibrCryptodome.Util.py3compatrrrrrr r#mainr r r r s    &