Merge 26c3070 into 3cbcaf0

JuliaStats · Dec 17, 2017 · a7b891c · a7b891c
2 parents 3cbcaf0 + 26c3070
commit a7b891c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/test/rossi.csv → test/data/rossi.csv b/test/rossi.csv → test/data/rossi.csv
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -146,13 +146,24 @@ end
 end
 
 @testset "Cox" begin
-    filepath = joinpath(@__DIR__, "rossi.csv")
+    filepath = joinpath(@__DIR__, "data", "rossi.csv")
     rossi = CSV.read(filepath)
     rossi[:event] = EventTime.(rossi[:week],rossi[:arrest] .== 1)
 
     outcome = coxph(@formula(event ~ fin+age+race+wexp+mar+paro+prio), rossi; tol = 1e-8)
     outcome_coefmat = coeftable(outcome)
 
+    @test sprint(show, outcome_coefmat) == """
+                Estimate Std.Error   z value Pr(>|z|)
+        fin    -0.379422  0.191379  -1.98256   0.0474
+        age   -0.0574377 0.0219995  -2.61087   0.0090
+        race      0.3139  0.307993   1.01918   0.3081
+        wexp   -0.149796  0.212224 -0.705837   0.4803
+        mar    -0.433704  0.381868  -1.13574   0.2561
+        paro  -0.0848711  0.195757 -0.433554   0.6646
+        prio   0.0914971 0.0286485   3.19378   0.0014
+        """
+
     coef_matrix = ModelMatrix(ModelFrame(@formula(event ~ 0+fin+age+race+wexp+mar+paro+prio), rossi)).m
     outcome_from_matrix     = coxph(coef_matrix, rossi[:event]; tol = 1e-8, l2_cost = 0)
     outcome_from_matrix32   = coxph(Float32.(coef_matrix), rossi[:event]; tol = 1e-5)
@@ -220,4 +231,4 @@ end
     end
     @test_throws ErrorException Survival.newton_raphson(wrong_fgh!, [2.2])
 
-end
+end