diff options
Diffstat (limited to 'tests/test_bio_membuf.py')
-rw-r--r-- | tests/test_bio_membuf.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_bio_membuf.py b/tests/test_bio_membuf.py index 49a0e24..ebac7b7 100644 --- a/tests/test_bio_membuf.py +++ b/tests/test_bio_membuf.py @@ -8,7 +8,7 @@ import os import multiprocessing from M2Crypto.BIO import MemoryBuffer -from tests import unittest +from . import unittest class TimeLimitExpired(Exception): @@ -110,7 +110,7 @@ def run_test(*args, **kwargs): assert mb.readlines() == [b'hello' + sep, b'world' + sep] def suite(): - return unittest.makeSuite(MemoryBufferTestCase) + return unittest.TestLoader().loadTestsFromTestCase(MemoryBufferTestCase) if __name__ == '__main__': |