mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-09-21 00:24:37 +03:00
60f372fcaf
* Bump clvm_tools_rs version for clvm stepper and add a test * lint * lint * flake8 * lint * lint? * formatting * More formatting * More formatting * hopefully final lint * A bit more formal python style for the clvm test * Add typing imports
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from typing import Any, Optional
|
|
from unittest import TestCase
|
|
|
|
from clvm_tools_rs import start_clvm_program
|
|
|
|
factorial = (
|
|
"ff02ffff01ff02ff02ffff04ff02ffff04ff05ff80808080ffff04ffff01ff02"
|
|
+ "ffff03ffff09ff05ffff010180ffff01ff0101ffff01ff12ff05ffff02ff02ff"
|
|
+ "ff04ff02ffff04ffff11ff05ffff010180ff808080808080ff0180ff018080"
|
|
)
|
|
|
|
factorial_function_hash = "de3687023fa0a095d65396f59415a859dd46fc84ed00504bf4c9724fca08c9de"
|
|
factorial_sym = {factorial_function_hash: "factorial"}
|
|
|
|
|
|
class TestRunProgram(TestCase):
|
|
def test_simple_program_run(self) -> None:
|
|
p = start_clvm_program(factorial, "ff0580", factorial_sym)
|
|
|
|
last: Optional[Any] = None
|
|
location: Optional[Any] = None
|
|
|
|
while not p.is_ended():
|
|
step_result = p.step()
|
|
if step_result is not None:
|
|
last = step_result
|
|
self.assertTrue("Failure" not in last)
|
|
|
|
if "Operator-Location" in last:
|
|
location = last["Operator-Location"]
|
|
|
|
self.assertTrue(last is not None)
|
|
self.assertTrue(location is not None)
|
|
if last is not None and location is not None:
|
|
self.assertTrue("Final" in last)
|
|
self.assertEqual(int(last["Final"]), 120)
|
|
self.assertTrue(location.startswith("factorial"))
|