diff --git a/test/manifolds/power_manifold.jl b/test/manifolds/power_manifold.jl index 0b7f5d7e1..2e30aee84 100644 --- a/test/manifolds/power_manifold.jl +++ b/test/manifolds/power_manifold.jl @@ -139,8 +139,13 @@ end ) push!(tvector_distributions_Mrn2, power_rn2_tv_dist) - Test.@test Distributions.support(power_s1_pt_dist) isa PowerPointDistribution - Test.@test Distributions.support(power_s1_tv_dist) isa PowerFVectorDistribution + MPointSupport = + Base.get_extension(Manifolds, :ManifoldsDistributionsExt).MPointSupport + FVectorSupport = + Base.get_extension(Manifolds, :ManifoldsDistributionsExt).FVectorSupport + + Test.@test Distributions.support(power_s1_pt_dist) isa MPointSupport + Test.@test Distributions.support(power_s1_tv_dist) isa FVectorSupport end @testset "get_component, set_component!, getindex and setindex!" begin diff --git a/test/manifolds/product_manifold.jl b/test/manifolds/product_manifold.jl index 61ae5e7ca..eb548a2ee 100644 --- a/test/manifolds/product_manifold.jl +++ b/test/manifolds/product_manifold.jl @@ -165,10 +165,14 @@ using RecursiveArrayTools: ArrayPartition ProductFVectorDistribution(distr_tv_M1, distr_tv_M2), ) - Test.@test Distributions.support(Mse_point_distributions[1]) isa - ProductPointDistribution + MPointSupport = + Base.get_extension(Manifolds, :ManifoldsDistributionsExt).MPointSupport + FVectorSupport = + Base.get_extension(Manifolds, :ManifoldsDistributionsExt).FVectorSupport + + Test.@test Distributions.support(Mse_point_distributions[1]) isa MPointSupport Test.@test Distributions.support(Mse_tvector_distributions[1]) isa - ProductFVectorDistribution + FVectorSupport end test_manifold(