runtests.jl

using Bijectors

using ChainRulesTestUtils
using Combinatorics
using DistributionsAD
using Documenter: Documenter
using Enzyme
using EnzymeTestUtils
using FiniteDifferences
using ForwardDiff
using Functors
using LogExpFunctions
using Mooncake
using ReverseDiff
using Tracker
using Zygote

using Random, LinearAlgebra, Test

using Bijectors:
    Shift,
    Scale,
    Logit,
    SimplexBijector,
    PDBijector,
    Permute,
    PlanarLayer,
    RadialLayer,
    Stacked,
    TruncatedBijector

using ChangesOfVariables: ChangesOfVariables
using InverseFunctions: InverseFunctions
using LazyArrays: LazyArrays

const GROUP = get(ENV, "GROUP", "All")

# Always include this since it can be useful for other tests.
include("ad/utils.jl")
include("bijectors/utils.jl")

if GROUP == "All" || GROUP == "Interface"
    include("interface.jl")
    include("transform.jl")
    include("norm_flows.jl")
    include("bijectors/permute.jl")
    include("bijectors/rational_quadratic_spline.jl")
    include("bijectors/named_bijector.jl")
    include("bijectors/leaky_relu.jl")
    include("bijectors/coupling.jl")
    include("bijectors/ordered.jl")
    include("bijectors/pd.jl")
    include("bijectors/reshape.jl")
    include("bijectors/corr.jl")
    include("bijectors/product_bijector.jl")

    include("distributionsad.jl")

    @testset "doctests" begin
        Documenter.DocMeta.setdocmeta!(
            Bijectors, :DocTestSetup, :(using Bijectors); recursive=true
        )
        doctestfilters = [
            # Ignore the source of a warning in the doctest output, since this is dependent
            # on host. This is a line that starts with "└ @ " and ends with the line number.
            r"└ @ .+:[0-9]+",
        ]
        Documenter.doctest(Bijectors; manual=false, doctestfilters=doctestfilters)
    end
end

if GROUP == "All" || GROUP == "AD"
    include("ad/chainrules.jl")
    include("ad/enzyme.jl")
    include("ad/flows.jl")
    include("ad/pd.jl")
    include("ad/corr.jl")
    include("ad/stacked.jl")
end