diff options
Diffstat (limited to 'tests/test_bio.py')
-rw-r--r-- | tests/test_bio.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/tests/test_bio.py b/tests/test_bio.py index d2b4fa4..89b2fc3 100644 --- a/tests/test_bio.py +++ b/tests/test_bio.py @@ -12,8 +12,8 @@ Author: Heikki Toivonen import logging from M2Crypto import BIO, Rand -from tests import unittest -from tests.fips import fips_mode +from . import unittest +from .fips import fips_mode log = logging.getLogger('test_bio') @@ -84,9 +84,8 @@ class CipherStreamTestCase(unittest.TestCase): def suite(): - t_suite = unittest.TestSuite() - t_suite.addTest(unittest.makeSuite(CipherStreamTestCase)) - return t_suite + return unittest.TestLoader().loadTestsFromTestCase(CipherStreamTestCase) + if __name__ == '__main__': Rand.load_file('randpool.dat', -1) |