delegate coefnames for wrapped models to model frame

JuliaStats · May 18, 2018 · 38c33ee · 38c33ee
1 parent fdeb053
commit 38c33ee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/statsmodel.jl b/src/statsmodel.jl
@@ -103,7 +103,7 @@ function StatsBase.predict(mm::DataFrameRegressionModel, df::AbstractDataFrame;
     return(out)
 end
 
-
+StatsBase.coefnames(model::DataFrameModels) = coefnames(model.mf)
 
 # coeftable implementation
 function StatsBase.coeftable(model::DataFrameModels)

diff --git a/test/statsmodel.jl b/test/statsmodel.jl
@@ -62,6 +62,9 @@ Base.show(io::IO, m::DummyModTwo) = println(io, m.msg)
     m = fit(DummyMod, f, d)
     @test model_response(m) == Array(d[:y])
 
+    ## coefnames delegated to model frame by default
+    @test coefnames(m) == coefnames(ModelFrame(f, d)) == ["(Intercept)", "x1", "x2", "x1 & x2"]
+
     ## test prediction method
     ## vanilla
     @test predict(m) == [ ones(size(d,1)) Array(d[:x1]) Array(d[:x2]) Array(d[:x1]).*Array(d[:x2]) ] * collect(1:4)