Spaces:
Sleeping
Sleeping
| 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 | |