test suite: use mmgen.pyversion for version checking

This commit is contained in:
The MMGen Project 2023-05-17 15:44:11 +00:00
commit 0fa5e707a9
Signed by: mmgen
GPG key ID: 3F8B1861E32B7DA2
2 changed files with 6 additions and 8 deletions

View file

@ -108,10 +108,9 @@ class TestKeccak(TestHashFunc):
def __init__(self):
from mmgen.contrib.keccak import keccak_256
self.t_cls = keccak_256
import platform
major,minor,_ = [int(s) for s in platform.python_version_tuple()]
if major > 3 or (major == 3 and minor >= 11):
ymsg(f'Skipping keccak random data test for Python version {major}.{minor} (no pysha3)')
from mmgen.pyversion import python_version
if python_version >= '3.11':
ymsg(f'Skipping keccak random data test for Python version {python_version} (no pysha3)')
self.hashlib = None
else:
import sha3

View file

@ -29,10 +29,9 @@ class unit_tests:
return True
def pysha3(self,name,ut): # ETH,XMR
import platform
major,minor,_ = [int(s) for s in platform.python_version_tuple()]
if major > 3 or (major == 3 and minor >= 11):
ymsg(f'Skipping pysha3 for Python version {major}.{minor}')
from mmgen.pyversion import python_version
if python_version >= '3.11':
ymsg(f'Skipping pysha3 for Python version {python_version}')
else:
from sha3 import keccak_256
return True