tests/test_pseudocode.py

from arena.optimizers.registry import get_example
from arena.pseudocode.renderer import generate_pseudocode


def test_pseudocode_contains_optimizer_and_update():
    code = get_example("Gradient Descent").code
    pseudo = generate_pseudocode(code)
    assert "Optimizer: `GradientDescent`" in pseudo
    assert "gradient descent update" in pseudo
    assert "θ_{t+1}" in pseudo


def test_generate_pseudocode_no_markdown_visitor_nameerror():
    code = get_example("Adam From Scratch").code
    pseudo = generate_pseudocode(code)
    assert "Optimizer: `AdamFromScratch`" in pseudo
    assert "Adam update" in pseudo