checkout: impl test generation

Author: sycai

packages/bigframes/scripts/generate_bigframes_bigquery/data_models.py

@@ -73,6 +73,7 @@ class BQFunc:
 class BQModule:
     yaml_file: pathlib.Path
     functions: list[BQFunc]
+    is_global: bool
 
     @property
     def module_path(self):

packages/bigframes/scripts/generate_bigframes_bigquery/file_generator.py

@@ -75,13 +75,20 @@ def _generate_accesor():
 
 
 def _generate_tests(bq_module: data_models.BQModule):
-    pass
+    content = template_renderer.render_tests(bq_module)
+
+    output_file = constants.TEST_OUTPUT_DIR.joinpath(
+        bq_module.module_path.with_name(f"test_{bq_module.module_path.name}")
+    ).with_suffix(".py")
+
+    _write_file(content, output_file, constants.TEST_OUTPUT_DIR.parent)
 
 
 def generate(bq_modules: list[data_models.BQModule]):
 
     for bq_module in bq_modules:
         _generate_op_defs(bq_module)
+        _generate_tests(bq_module)
 
     # Ruff format
     _run_ruff()

packages/bigframes/scripts/generate_bigframes_bigquery/template_renderer.py

@@ -230,4 +230,18 @@ def render_operation(
 
 def render_tests(bq_module: data_models.BQModule) -> str:
 
-    pass
+    import_path = "bigframes.operations.googlesql." + ".".join(
+        bq_module.module_path.parts
+    )
+    functions = []
+    for bq_func in bq_module.functions:
+        functions.append(_to_bigframes_func(bq_func))
+
+    return constants.TEMPLATES["test_operation"].render(
+        yaml_path=bq_module.yaml_file.relative_to(constants.PACKAGE_ROOT),
+        script_path=constants.SCRIPT_PATH_RELATIVE,
+        import_path=import_path,
+        short_name=bq_module.module_path.name,
+        is_global=bq_module.is_global,
+        functions=functions,
+    )

packages/bigframes/scripts/generate_bigframes_bigquery/yaml_parser.py

@@ -48,11 +48,11 @@ def _parse_func_impl(impl_data: Any) -> data_models.BQFuncImpl:
     )
 
 
-def _parse_bq_func(func_data: Any, module_path: pathlib.Path) -> data_models.BQFunc:
+def _parse_bq_func(
+    func_data: Any, module_name: str, is_global: bool
+) -> data_models.BQFunc:
     op_base_name = _to_snake_case(func_data["name"])
 
-    module_name = module_path.name
-    is_global = "global_namespace" in module_path.parts
     if not is_global and op_base_name.startswith(module_name + "_"):
         op_base_name = op_base_name[len(module_name) + 1 :]
 
@@ -72,13 +72,15 @@ def parse_yaml(yaml_file: pathlib.Path) -> data_models.BQModule:
 
     functions = []
     module_path = yaml_file.relative_to(constants.DATA_DIR).with_suffix("")
+    is_global = "global_namespace" in module_path.parts
     if isinstance(data, dict) and "scalar_functions" in data:
         functions = [
-            _parse_bq_func(func_data, module_path)
+            _parse_bq_func(func_data, module_path.name, is_global)
             for func_data in data["scalar_functions"]
         ]
 
     return data_models.BQModule(
         yaml_file=yaml_file,
         functions=functions,
+        is_global=is_global,
     )