From 851b146f541724eda1caf6076fa9728387ded0d4 Mon Sep 17 00:00:00 2001
From: Jose Esparza <28990958+pebeto@users.noreply.github.com>
Date: Fri, 18 Aug 2023 02:04:28 -0500
Subject: [PATCH] Adding aditional test case on save

---
 test/base.jl | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/test/base.jl b/test/base.jl
index d1be5e1..8c4796f 100644
--- 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