merge test

joshday · Jun 28, 2018 · 809adee · 809adee
1 parent 521f523
commit 809adee
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,13 +1,14 @@
 using Compat, Compat.Test, OnlineStatsBase, LearnBase
 
-import OnlineStatsBase: OnlineStat, _fit!, fit!,
+import OnlineStatsBase: OnlineStat, _fit!, fit!, _merge!
     EqualWeight, ExponentialWeight, LearningRate, LearningRate2, HarmonicWeight,
     McclainWeight
 #-----------------------------------------------------------------------#
 mutable struct FakeStat <: OnlineStat{Number}
     n::Int 
 end 
 _fit!(o::FakeStat, y) = (o.n += 1)
+_merge!(o::FakeStat, o2::FakeStat) = (o.n += o2.n)
 
 struct FakeStat2 <: OnlineStat{Number} end 
 println(FakeStat(0))
@@ -21,6 +22,7 @@ println(FakeStat(0))
     @test FakeStat(10) == FakeStat(10)
     @test FakeStat(0) != FakeStat2()
     @test value(fit!(FakeStat(0), rand(100))) == 100
+    @test merge(FakeStat(10), FakeStat(11)).n == 21
 end
 
 #-----------------------------------------------------------------------# Weight