reasoning-gym/reasoning_gym/algebra/test.py