From 56777c94c938fb51852ee085c6886ce047fd2660 Mon Sep 17 00:00:00 2001
From: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Date: Fri, 26 Jan 2024 11:32:24 +0000
Subject: [PATCH] added regression test

---
 test/variational/advi.jl | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/test/variational/advi.jl b/test/variational/advi.jl
index ecca26545..17cbc7ea2 100644
--- a/test/variational/advi.jl
+++ b/test/variational/advi.jl
@@ -64,5 +64,10 @@
         x0_inv = inverse(b)(z0)
         @test size(x0_inv) == size(x0)
         @test all(x0 .≈ x0_inv)
+
+        # And regression for https://github.com/TuringLang/Turing.jl/issues/2160.
+        q = vi(m, ADVI(10, 1000))
+        x = rand(q, 1000)
+        @test mean(eachcol(x)) ≈ [0.5, 0.5] atol=0.1
     end
 end