get_builtins_initial_stack.py

import cairo_vm_py

def new_runner(program_name: str):
    with open(f"cairo_programs/{program_name}.json") as file:
        return cairo_vm_py.CairoRunner(file.read(), "main", "all_cairo", False)

def test_program(program_name: str):
    runner = new_runner(program_name)
    
    builtins_initial_stack = runner.get_program_builtins_initial_stack()
    assert builtins_initial_stack == [], 'Initial stack should be empty.'
    
    runner.cairo_run(False)
    
    builtins_final_stack = runner.get_program_builtins_initial_stack()
    
    expected_output = [cairo_vm_py.RelocatableValue((2,0))]
    assert builtins_final_stack == expected_output

if __name__ == "__main__":
    test_program("get_builtins_initial_stack")
    print("\nget_builtins_initial_stack test passed.")