test: add PromoteTypeConverter tests

ReactiveBayes · Oct 10, 2023 · e9d869a · e9d869a
1 parent 11b8b2b
commit e9d869a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/test/distributions_setuptests.jl b/test/distributions_setuptests.jl
@@ -11,7 +11,8 @@ import ExponentialFamily:
     promote_samplefloattype,
     paramfloattype,
     convert_paramfloattype,
-    FactorizedJoint
+    FactorizedJoint,
+    PromoteTypeConverter
 
 function generate_random_distributions(::Type{V} = Any; seed = abs(rand(Int)), Types = (Float32, Float64)) where {V}
     rng = StableRNG(seed)

diff --git a/test/distributions_tests.jl b/test/distributions_tests.jl
@@ -154,3 +154,13 @@ end
         )
     end
 end
+
+@testitem "TypeConverter" begin
+    include("./distributions_setuptests.jl")
+
+    for original_T in (Float16, Float32, Float64), target_T in (Float16, Float32, Float64), n in (1, 2, 3)
+        converter = PromoteTypeConverter(target_T, convert)
+
+        @test typeof(@inferred(converter(rand(original_T)))) === target_T
+    end
+end