mirror of
https://github.com/tig-pool-nk/tig-monorepo.git
synced 2026-03-05 14:57:22 +08:00
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
import unittest
|
|
import sys
|
|
import os
|
|
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
|
|
from common.utils import u64s_from_str, u8s_from_str, jsonify
|
|
from common.merkle_tree import MerkleHash
|
|
from common.structs import BenchmarkSettings, OutputData
|
|
|
|
class TestData(unittest.TestCase):
|
|
def test_calc_solution_signature(self):
|
|
output_data = OutputData(
|
|
nonce=123,
|
|
runtime_signature=456,
|
|
fuel_consumed=789,
|
|
solution="test",
|
|
cpu_arch="AMD64"
|
|
)
|
|
|
|
# Assert same as Rust version: tig-structs/tests/core.rs
|
|
self.assertEqual(output_data.calc_solution_signature(), 11204800550749450632)
|
|
|
|
def test_calc_seed(self):
|
|
settings = BenchmarkSettings(
|
|
player_id="some_player",
|
|
block_id="some_block",
|
|
challenge_id="some_challenge",
|
|
algorithm_id="some_algorithm",
|
|
difficulty=[1, 2, 3]
|
|
)
|
|
|
|
rand_hash = "random_hash"
|
|
nonce = 1337
|
|
|
|
# Assert same as Rust version: tig-structs/tests/core.rs
|
|
expected = bytes([
|
|
135, 168, 152, 35, 57, 28, 184, 91, 10, 189, 139, 111, 171, 82, 156, 14,
|
|
165, 68, 80, 41, 169, 236, 42, 41, 198, 73, 124, 78, 130, 216, 168, 67
|
|
])
|
|
self.assertEqual(settings.calc_seed(rand_hash, nonce), expected)
|
|
|
|
def test_outputdata_to_merklehash(self):
|
|
output_data = OutputData(
|
|
nonce=123,
|
|
runtime_signature=456,
|
|
fuel_consumed=789,
|
|
solution="test",
|
|
cpu_arch="AMD64"
|
|
)
|
|
|
|
merkle_hash = output_data.to_merkle_hash()
|
|
|
|
# Assert same as Rust version: tig-structs/tests/core.rs
|
|
expected = MerkleHash(bytes([
|
|
79, 126, 186, 90, 12, 111, 100, 8, 120, 150, 225, 176, 200, 201, 125, 150, 58, 122,
|
|
214, 216, 68, 6, 125, 247, 248, 4, 165, 185, 157, 44, 13, 151
|
|
]))
|
|
self.assertEqual(merkle_hash, expected)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |