Adding aditional test case on save

JuliaAI · Aug 18, 2023 · 851b146 · 851b146
1 parent 993f3d1
commit 851b146
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/base.jl b/test/base.jl
@@ -33,6 +33,12 @@
 
  loaded_mach = machine(artifacts[1].filepath)
  @test loaded_mach.model isa ProbabilisticPipeline
+
+ test_x, test_y = make_moons(1)
+ pred = predict(mach, test_x)[1]
+ loaded_mach_pred = predict(loaded_mach, test_x)[1]
+ @test pdf(pred, 0) == pdf(loaded_mach_pred, 0)
+ @test pdf(pred, 1) == pdf(loaded_mach_pred, 1)
  end
 
  @testset "accesor methods" begin