added a test for RQS to interface.jl but should move away from this

TuringLang · Sep 12, 2020 · 160f44b · 160f44b
1 parent 5d0ce8f
commit 160f44b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/interface.jl b/test/interface.jl
@@ -7,7 +7,7 @@ using Tracker
 using DistributionsAD
 
 using Bijectors
-using Bijectors: Log, Exp, Shift, Scale, Logit, SimplexBijector, PDBijector, Permute, PlanarLayer, RadialLayer, Stacked, TruncatedBijector, ADBijector
+using Bijectors: Log, Exp, Shift, Scale, Logit, SimplexBijector, PDBijector, Permute, PlanarLayer, RadialLayer, Stacked, TruncatedBijector, ADBijector, RationalQuadraticSpline
 
 Random.seed!(123)
 
@@ -159,7 +159,8 @@ end
         (SimplexBijector(), mapslices(z -> normalize(z, 1), rand(2, 3); dims = 1)),
         (stack(Exp{0}(), Scale(2.0)), randn(2, 3)),
         (Stacked((Exp{1}(), SimplexBijector()), [1:1, 2:3]),
-         mapslices(z -> normalize(z, 1), rand(3, 2); dims = 1))
+         mapslices(z -> normalize(z, 1), rand(3, 2); dims = 1)),
+        (RationalQuadraticSpline(randn(3), randn(3), randn(3 - 1), 2.), [-0.5, 0.5])
     ]
 
     for (b, xs) in bs_xs