[Tests] Improve tests

Fixes #43
JuliaSurv · May 9, 2024 · 9efbdab · 9efbdab
1 parent 4ce0824
commit 9efbdab
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/test/sampletest.jl b/test/sampletest.jl
@@ -31,17 +31,13 @@ end
     using DataFrames
     using RateTables
 
-    colrec.country = rand(keys(hmd_countries),nrow(colrec))
-    fit(PoharPerme, @formula(Surv(time,status)~1), colrec, frpop)
-    fit(EdererI, @formula(Surv(time,status)~sex), colrec, frpop)
-    fit(EdererII, @formula(Surv(time,status)~sex), colrec, frpop)
-    fit(Hakulinen, @formula(Surv(time,status)~sex), colrec, frpop)
-    fit(GraffeoTest, @formula(Surv(time,status)~stage), colrec, frpop)
+    for E in (PoharPerme, EdererI, EdererII, Hakulinen)
+        npe = fit(E, @formula(Surv(time,status)~1), colrec, frpop)
+        @assert !any(isnan.(npe.Sₑ..., npe.∂Λₑ..., npe.σₑ))
+    end
+
     fit(GraffeoTest, @formula(Surv(time,status)~stage+Strata(sex)), colrec, frpop)
-
     @test true
-
-    # How to use R and R packages in here ? 
 end