summaryrefslogtreecommitdiff
path: root/lib/Crypto/SelfTest/PublicKey/test_DSA.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/Crypto/SelfTest/PublicKey/test_DSA.py')
-rw-r--r--lib/Crypto/SelfTest/PublicKey/test_DSA.py12
1 files changed, 2 insertions, 10 deletions
diff --git a/lib/Crypto/SelfTest/PublicKey/test_DSA.py b/lib/Crypto/SelfTest/PublicKey/test_DSA.py
index 72b9e95..037e087 100644
--- a/lib/Crypto/SelfTest/PublicKey/test_DSA.py
+++ b/lib/Crypto/SelfTest/PublicKey/test_DSA.py
@@ -225,16 +225,8 @@ def get_tests(config={}):
from Crypto.PublicKey import _fastmath
tests += list_test_cases(DSAFastMathTest)
except ImportError:
- from distutils.sysconfig import get_config_var
- import inspect
- ext_suffix = get_config_var("EXT_SUFFIX") or get_config_var("SO")
- _fm_path = os.path.normpath(os.path.dirname(os.path.abspath(
- inspect.getfile(inspect.currentframe())))
- +"/../../PublicKey/_fastmath"+ext_suffix)
- if os.path.exists(_fm_path):
- raise ImportError("While the _fastmath module exists, importing "+
- "it failed. This may point to the gmp or mpir shared library "+
- "not being in the path. _fastmath was found at "+_fm_path)
+ from Crypto.SelfTest.st_common import handle_fastmath_import_error
+ handle_fastmath_import_error()
tests += list_test_cases(DSASlowMathTest)
return tests