| # -*- coding: utf-8 -*- | |
| # | |
| # SelfTest/st_common.py: Common functions for SelfTest modules | |
| # | |
| # Written in 2008 by Dwayne C. Litzenberger <[email protected]> | |
| # | |
| # =================================================================== | |
| # The contents of this file are dedicated to the public domain. To | |
| # the extent that dedication to the public domain is not available, | |
| # everyone is granted a worldwide, perpetual, royalty-free, | |
| # non-exclusive license to exercise all rights associated with the | |
| # contents of this file for any purpose whatsoever. | |
| # No rights are reserved. | |
| # | |
| # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | |
| # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |
| # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | |
| # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS | |
| # BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN | |
| # ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN | |
| # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
| # SOFTWARE. | |
| # =================================================================== | |
| """Common functions for SelfTest modules""" | |
| import unittest | |
| import binascii | |
| from Crypto.Util.py3compat import b | |
| def list_test_cases(class_): | |
| """Return a list of TestCase instances given a TestCase class | |
| This is useful when you have defined test* methods on your TestCase class. | |
| """ | |
| return unittest.TestLoader().loadTestsFromTestCase(class_) | |
| def strip_whitespace(s): | |
| """Remove whitespace from a text or byte string""" | |
| if isinstance(s,str): | |
| return b("".join(s.split())) | |
| else: | |
| return b("").join(s.split()) | |
| def a2b_hex(s): | |
| """Convert hexadecimal to binary, ignoring whitespace""" | |
| return binascii.a2b_hex(strip_whitespace(s)) | |
| def b2a_hex(s): | |
| """Convert binary to hexadecimal""" | |
| # For completeness | |
| return binascii.b2a_hex(s) | |
| # vim:set ts=4 sw=4 sts=4 expandtab: | |