diff --git a/test/BetaSchedules.jl b/test/BetaSchedules.jl new file mode 100644 index 0000000..82dc8d1 --- /dev/null +++ b/test/BetaSchedules.jl @@ -0,0 +1,18 @@ +using Diffusers.BetaSchedules +using Test + +@testset "Variance schedules tests" begin + @testset "β increases monotonically" begin + T = 1000 + + β_linear = linear_beta_schedule(T) + β_scaled_linear = scaled_linear_beta_schedule(T) + β_cosine = cosine_beta_schedule(T) + β_sigmoid = sigmoid_beta_schedule(T) + + @test all(diff(β_linear) .>= 0) + @test all(diff(β_scaled_linear) .>= 0) + @test all(diff(β_cosine) .>= 0) + @test all(diff(β_sigmoid) .>= 0) + end +end diff --git a/test/runtests.jl b/test/runtests.jl index 6cc9197..3a7ab0d 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -2,3 +2,5 @@ using Diffusers using Aqua Aqua.test_all(Diffusers) + +include("BetaSchedules.jl")