Merge pull request #324 from gustaphe/customhead

Fix dataframe 'head' argument
korsbo · Dec 15, 2024 · 927c39e · 927c39e
2 parents ad0ab40 + 22d4dab
commit 927c39e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/ext/DataFramesExt.jl b/ext/DataFramesExt.jl
@@ -5,7 +5,7 @@ isdefined(Base, :get_extension) ? (using DataFrames) : (using ..DataFrames)
 
 @latexrecipe function f(d::DataFrame)
     env --> :mdtable
-    head := propertynames(d)
+    head --> propertynames(d)
     if kwargs[:env] == :array
         return vcat(permutedims(propertynames(d)), Matrix(d))
     end

diff --git a/test/plugins/DataFrames_test.jl b/test/plugins/DataFrames_test.jl
@@ -54,3 +54,11 @@ y & 1//2\\
 z & 8\\
 \end{tabular}
 ", "\r\n"=>"\n")
+
+@test latexify(df; head=["x", "y"]) ==
+Markdown.md"|   x |                      y |
+| ---:| ----------------------:|
+| $x$ | $\frac{\alpha}{\beta}$ |
+| $y$ |          $\frac{1}{2}$ |
+| $z$ |                    $8$ |
+"