remove TestProcessProto import from test_openclpow module

This commit is contained in:
navjot 2021-02-01 15:05:04 +05:30
parent 448e9e2f36
commit f381721bec
No known key found for this signature in database
GPG Key ID: 9EE70AFD71357F1C

View File

@ -4,14 +4,12 @@ Tests for openclpow module
import hashlib import hashlib
import unittest import unittest
import pybitmessage.openclpow as openclpow
from struct import pack, unpack from struct import pack, unpack
from test_process import TestProcessProto from pybitmessage import openclpow
class TestOpenClPow(TestProcessProto): class TestOpenClPow(unittest.TestCase):
""" """
Main opencl test case Main opencl test case
""" """
@ -19,16 +17,17 @@ class TestOpenClPow(TestProcessProto):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
openclpow.initCL() openclpow.initCL()
super(TestOpenClPow, cls).setUpClass()
@unittest.skipIf(not openclpow.enabledGpus, @unittest.skipUnless(openclpow.enabledGpus, "No GPUs found / enabled")
"No GPUs found / enabled")
def test_openclpow(self): def test_openclpow(self):
"""Check the working of openclpow module""" """Check the working of openclpow module"""
target_ = 54227212183 target_ = 54227212183
initialHash = ("3758f55b5a8d902fd3597e4ce6a2d3f23daff735f65d9698c270987f4e67ad590" initialHash = (
"b93f3ffeba0ef2fd08a8dc2f87b68ae5a0dc819ab57f22ad2c4c9c8618a43b3").decode("hex") "3758f55b5a8d902fd3597e4ce6a2d3f23daff735f65d9698c270987f4e67ad590"
"b93f3ffeba0ef2fd08a8dc2f87b68ae5a0dc819ab57f22ad2c4c9c8618a43b3"
).decode("hex")
nonce = openclpow.do_opencl_pow(initialHash.encode("hex"), target_) nonce = openclpow.do_opencl_pow(initialHash.encode("hex"), target_)
trialValue, = unpack( trialValue, = unpack(
'>Q', hashlib.sha512(hashlib.sha512(pack('>Q', nonce) + initialHash).digest()).digest()[0:8]) '>Q', hashlib.sha512(hashlib.sha512(
pack('>Q', nonce) + initialHash).digest()).digest()[0:8])
self.assertLess((nonce - trialValue), target_) self.assertLess((nonce - trialValue), target_)