WIP. Test gen doc from macros

comphead · Oct 8, 2024 · 5fe25f3 · 5fe25f3
1 parent dca1a33
commit 5fe25f3
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/datafusion/functions/src/math/log.rs b/datafusion/functions/src/math/log.rs
@@ -38,7 +38,7 @@ use datafusion_expr::{
 };
 use datafusion_expr::{ScalarUDFImpl, Signature, Volatility};
 
-#[udf_doc(description="log_description", example="log_example")]
+#[udf_doc(description = "log_description", example = "log_example")]
 #[derive(Debug)]
 pub struct LogFunc {
     signature: Signature,
@@ -475,9 +475,12 @@ mod tests {
     #[test]
     fn test_doc() {
         let log = LogFunc::new();
-        assert_eq!(log.documentation_test(), Some(DocumentationTest {
-            description: "log_description".to_string(),
-            syntax_example: "log_example".to_string(),
-        }));
+        assert_eq!(
+            log.documentation_test(),
+            Some(DocumentationTest {
+                description: "log_description".to_string(),
+                syntax_example: "log_example".to_string(),
+            })
+        );
     }
 }
diff --git a/datafusion/macros/src/lib.rs b/datafusion/macros/src/lib.rs
@@ -1,8 +1,8 @@
 extern crate proc_macro;
+use datafusion_pre_macros::DocumentationTest;
 use proc_macro::TokenStream;
 use quote::quote;
 use syn::{parse_macro_input, DeriveInput, Lit, LitStr, Meta, MetaNameValue};
-use datafusion_pre_macros::DocumentationTest;
 
 #[proc_macro_attribute]
 pub fn udf_doc(args: TokenStream, input: TokenStream) -> TokenStream {
@@ -29,7 +29,6 @@ pub fn udf_doc(args: TokenStream, input: TokenStream) -> TokenStream {
 
     //eprintln!("input={input:?}");
 
-
     let expanded = quote! {
         #input
 

diff --git a/datafusion/pre-macros/src/lib.rs b/datafusion/pre-macros/src/lib.rs
@@ -4,4 +4,4 @@ pub struct DocumentationTest {
     pub description: String,
     /// a brief example of the syntax. For example "ascii(str)"
     pub syntax_example: String,
-}
+}