From 6034d4172d0aa1d1b254e49657c5c1b43b649f45 Mon Sep 17 00:00:00 2001
From: iomgaa <iomgaaycz@gmail.com>
Date: Sat, 11 Jul 2026 08:21:06 -0400
Subject: [PATCH] test(question_gen): update __all__ assertion to include
 synthesizer exports

---
 tests/unit/test_question_gen_api.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/tests/unit/test_question_gen_api.py b/tests/unit/test_question_gen_api.py
index 5a4a8f0..361555f 100644
--- a/tests/unit/test_question_gen_api.py
+++ b/tests/unit/test_question_gen_api.py
@@ -35,6 +35,13 @@ class TestQuestionGenPublicAPI:
         assert hasattr(mod, "stratified_sample")
 
     def test_all_exports(self) -> None:
-        """__all__ 包含预期的公开 API。"""
+        """__all__ 包含预期的公开 API（loader + synthesizer）。"""
         mod = importlib.import_module("app.question_gen")
-        assert set(mod.__all__) == {"load_benchmark", "stratified_sample"}
+        assert set(mod.__all__) == {
+            "load_benchmark",
+            "stratified_sample",
+            "TASK_TYPE_LEVEL_MAP",
+            "AnchorContext",
+            "generate_one",
+            "sample_anchor",
+        }