From f381721bec31641002e2f240309600c4994855a7 Mon Sep 17 00:00:00 2001 From: navjot Date: Mon, 1 Feb 2021 15:05:04 +0530 Subject: [PATCH] remove TestProcessProto import from test_openclpow module --- src/tests/test_openclpow.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/src/tests/test_openclpow.py b/src/tests/test_openclpow.py index e98b7362..63551ee2 100644 --- a/src/tests/test_openclpow.py +++ b/src/tests/test_openclpow.py @@ -4,14 +4,12 @@ Tests for openclpow module import hashlib import unittest -import pybitmessage.openclpow as openclpow - 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 """ @@ -19,16 +17,17 @@ class TestOpenClPow(TestProcessProto): @classmethod def setUpClass(cls): openclpow.initCL() - super(TestOpenClPow, cls).setUpClass() - @unittest.skipIf(not openclpow.enabledGpus, - "No GPUs found / enabled") + @unittest.skipUnless(openclpow.enabledGpus, "No GPUs found / enabled") def test_openclpow(self): """Check the working of openclpow module""" target_ = 54227212183 - initialHash = ("3758f55b5a8d902fd3597e4ce6a2d3f23daff735f65d9698c270987f4e67ad590" - "b93f3ffeba0ef2fd08a8dc2f87b68ae5a0dc819ab57f22ad2c4c9c8618a43b3").decode("hex") + initialHash = ( + "3758f55b5a8d902fd3597e4ce6a2d3f23daff735f65d9698c270987f4e67ad590" + "b93f3ffeba0ef2fd08a8dc2f87b68ae5a0dc819ab57f22ad2c4c9c8618a43b3" + ).decode("hex") nonce = openclpow.do_opencl_pow(initialHash.encode("hex"), target_) 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_)