import asyncio
import json
from sympy import symbols, Eq
from strategy_manager import StrategyManager

async def run_test():
    sm = StrategyManager()
    
    print("--- TEST 1: General Prompt with Good Math ---")
    topic_id = "CIRCLE_EQUATION"
    data_anchor = {"center": [0,0], "radius": 5}
    llm_resp = {"solution": "x^2 + y^2 = 25"}
    
    res = sm._validate(topic_id, llm_resp, data_anchor)
    print(f"Validation Result: {res}")
    
    print("\n--- TEST 2: General Prompt with Bad Math ---")
    llm_resp_bad = {"solution": "x^2 - (y^2 / 3) = 1"}
    res2 = sm._validate(topic_id, llm_resp_bad, data_anchor)
    print(f"Validation Result: {res2}")
    
    print("\n--- TEST 3: General Prompt on Unknown Topic ---")
    res3 = sm._validate("WEIRD_TOPIC", {}, {})
    print(f"Validation Result: {res3}")

if __name__ == "__main__":
    asyncio.run(run_test())