PyBitmessage-2025-02-05/src/tests/test_openclpow.py

"""
Tests for openclpow module
"""

import unittest

from pybitmessage import openclpow, proofofwork


class TestOpenClPow(unittest.TestCase):
    """
    Main opencl test case
    """

    @classmethod
    def setUpClass(cls):
        openclpow.initCL()

    @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")
        nonce = openclpow.do_opencl_pow(initialHash.encode("hex"), target_)
        self.assertLess(
            nonce - proofofwork.trial_value(nonce, initialHash), target_)