From 8dd1a712685b4e76e7f84484b7ae9ed2ab3c75ac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurent@fainsin.bzh>
Date: Sun, 6 Aug 2023 21:19:04 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20(BetaSchedules)=20add=20variance?=
 =?UTF-8?q?=20schedules=20monotonicity=20check?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 test/BetaSchedules.jl | 18 ++++++++++++++++++
 test/runtests.jl      |  2 ++
 2 files changed, 20 insertions(+)
 create mode 100644 test/BetaSchedules.jl

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")