# -*- coding: utf-8 -*- # # SelfTest/st_common.py: Common functions for SelfTest modules # # Written in 2008 by Dwayne C. Litzenberger # # =================================================================== # 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: