fix a few tests

JuliaManifolds · Aug 8, 2024 · 3715e6c · 3715e6c
1 parent b0742cc
commit 3715e6c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git 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
@@ -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(