1- const CIENV = get (ENV , " TRAVIS" , " " ) == " true" || get (ENV , " CI" , " " ) == " true"
2-
31using DocStringExtensions, LinearAlgebra, LogDensityProblems, OffsetArrays, Parameters,
42 Random, Test, TransformVariables, StaticArrays
53import Tracker, ForwardDiff
@@ -8,6 +6,8 @@ using TransformVariables:
86 AbstractTransform, ScalarTransform, VectorTransform, ArrayTransform,
97 unit_triangular_dimension, logistic, logistic_logjac, logit, inverse_and_logjac
108
9+ const CIENV = get (ENV , " CI" , " " ) == " true"
10+
1111include (" utilities.jl" )
1212
1313Random. seed! (1 )
8686 @testset " dimension checks" begin
8787 U = UnitVector (3 )
8888 x = zeros (3 ) # incorrect
89- @test_throws ArgumentError U (x)
9089 @test_throws ArgumentError transform (U, x)
9190 @test_throws ArgumentError transform_and_logjac (U, x)
9291 end
107106 @testset " dimension checks" begin
108107 S = UnitSimplex (3 )
109108 x = zeros (3 ) # incorrect
110- @test_throws ArgumentError S (x)
111109 @test_throws ArgumentError transform (S, x)
112110 @test_throws ArgumentError transform_and_logjac (S, x)
113111 end
131129 C = CorrCholeskyFactor (3 )
132130 wrong_x = zeros (dimension (C) + 1 )
133131
134- @test_throws ArgumentError C (wrong_x)
135132 @test_throws ArgumentError transform (C, wrong_x)
136133 @test_throws ArgumentError transform_and_logjac (C, wrong_x)
137134 end
282279 tt = as ((a = as (Tuple (as (Vector, asℝ₊, 2 ) for _ in 1 : N)),
283280 b = as (Tuple (UnitVector (n) for n in 1 : N))))
284281 x = randn (dimension (tt))
285- y = tt ( x)
282+ y = transform (tt, x)
286283 x′ = inverse (tt, y)
287284 @test x ≈ x′
288285 end
310307 f = as (Array, 9 )))
311308 z = zeros (dimension (t))
312309 f (θ) = θ. a + θ. b + sum (abs2, θ. c. d) + sum (abs2, θ. c. e)
313- @test (@inferred f (t ( z))) isa Float64
310+ @test (@inferred f (transform (t, z))) isa Float64
314311 @test (@inferred transform_logdensity (t, f, z)) isa Float64
315312end
316313
@@ -486,12 +483,12 @@ end
486483# ###
487484
488485@testset " broadcasting" begin
489- @test as𝕀 .( [0 , 0 ]) == [0.5 , 0.5 ]
486+ @test transform .(as𝕀, [0 , 0 ]) == [0.5 , 0.5 ]
490487
491488 t = UnitVector (3 )
492489 d = dimension (t)
493490 x = [zeros (d), zeros (d)]
494- @test t .( x) == map (t , x)
491+ @test transform .(t, x) == map (x -> transform (t, x) , x)
495492end
496493
497494# ###
0 commit comments