U g @srdZddlZddlZddlmZddlmZGdddeZdddZ ifd d Z e d krnd d Z ej d ddS)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 @/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/SelfTest/__init__.pyr&szSelfTestError.__init__N)__name__ __module__ __qualname__rr r r r r%src Ks|dkr i}t}|dkr:|dkr.t|d}||n$|dkrV||j|dntd|dkrrt|d<n||d<tjfd|i|}||}|s|dkrt 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 failed)unittest TestSuite get_testsZaddTests ValueErrorrZTextTestRunnerrunZ wasSuccessfulsysstderrwritegetvaluer) modulerrtestsrkwargssuiterunnerr r r r r,s(      rc CsDg}ddddddddd g }|D] }td |}||j|d 7}q|S) NZCipherZHashProtocolZ PublicKeyRandomZUtil SignatureIOZMathzCryptodome.SelfTest.r)rr)rrZ module_namesnamerr r r rRs r__main__cCs ttSr)rrrr r r r r bsr )Z defaultTest)NrNNN)__doc__rr importlibrZCryptodome.Util.py3compatrrrrrr r mainr r r r s   &