mirror of
https://github.com/Legrandin/pycryptodome.git
synced 2025-10-26 03:04:44 +00:00
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
# SelfTest/st_common.py: Common functions for SelfTest modules
|
||
|
|
#
|
||
|
|
# =======================================================================
|
||
|
|
# Copyright (C) 2008 Dwayne C. Litzenberger <dlitz@dlitz.net>
|
||
|
|
#
|
||
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
||
|
|
# a copy of this software and associated documentation files (the
|
||
|
|
# "Software"), to deal in the Software without restriction, including
|
||
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
||
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
||
|
|
# permit persons to whom the Software is furnished to do so.
|
||
|
|
#
|
||
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
# =======================================================================
|
||
|
|
#
|
||
|
|
|
||
|
|
"""Common functions for SelfTest modules"""
|
||
|
|
|
||
|
|
__revision__ = "$Id$"
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import string
|
||
|
|
import binascii
|
||
|
|
|
||
|
|
class _list_testloader(unittest.TestLoader):
|
||
|
|
suiteClass = list
|
||
|
|
|
||
|
|
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 _list_testloader().loadTestsFromTestCase(class_)
|
||
|
|
|
||
|
|
def strip_whitespace(s):
|
||
|
|
"""Remove whitespace from a string"""
|
||
|
|
table = string.maketrans(string.whitespace, " " * len(string.whitespace))
|
||
|
|
s = s.translate(table).replace(" ", "")
|
||
|
|
return s
|
||
|
|
|
||
|
|
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:
|