Codegen: move rust.doc_test_signature to parametrized pragmas

This commit is contained in:
Paolo Tranquilli
2024-09-20 09:53:57 +02:00
parent 70997e8189
commit 9d6ee09f65
4 changed files with 2 additions and 5 deletions

View File

@@ -55,7 +55,7 @@ def generate(opts, renderer):
continue
assert not adding_code, "Unterminated code block in docstring: " + "\n".join(cls.doc)
test_name = inflection.underscore(cls.name)
signature = cls.rust_doc_test_function
signature = cls.pragmas.get("rust_doc_test_signature", "() -> ()")
fn = signature and Function(f"test_{test_name}", signature)
if fn:
indent = 4 * " "