small stress testing

problem_generator.py

@@ -22,7 +22,6 @@ def generate_linear():
     c = a * ans + b
     x = symbols('x')
     expr = a * x + b
-    # expanded = n
     s = sstr(expr)
 
     return {
@@ -137,7 +136,7 @@ def generate_quadratic ():
 @register_problem_generator("difference_squares")
 def generate_difference_squares ():
     #x² - a² = 0
-    ans = random.choice([i for i in range(0, 13)])
+    ans = random.choice([i for i in range(0, 13) if i != 0])
     a = ans
 
     x = symbols('x')
@@ -178,7 +177,7 @@ def generate_radical ():
 
     x = symbols('x')
     expr = root(x+a, 2)
-
+    
     return {
         "type": "radical",
         "problem": f"{sstr(expr)} = {b}",
@@ -265,6 +264,7 @@ def generate_problem():
     
     problem_type = random.choices(types, weights=weights)[0]
     template = TEMPLATES[problem_type]
-    return generate_quadratic()
-    #return template()
+    #return list(TEMPLATES.values())[11]()
+    #return generate_quadratic()
+    return template()