make 'rand' of Tetrahedron consistent

JuliaReach · Jul 23, 2024 · db7b38b · db7b38b
1 parent fd1b70e
commit db7b38b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/Sets/Tetrahedron/rand.jl b/src/Sets/Tetrahedron/rand.jl
@@ -1,8 +1,9 @@
-function rand(::Type{Tetrahedron}; N::Type{<:Real}=Float64, rng::AbstractRNG=GLOBAL_RNG,
-              seed::Union{Int,Nothing}=nothing)
+function rand(::Type{Tetrahedron}; N::Type{<:Real}=Float64, dim::Int=3,
+              rng::AbstractRNG=GLOBAL_RNG, seed::Union{Int,Nothing}=nothing)
+    @assert dim == 3 "cannot create a random Tetrahedron of dimension $dim"
     require(@__MODULE__, :LazySets; fun_name="rand")
 
+    rng = reseed!(rng, seed)
     P = rand(VPolytope; N=N, dim=3, rng=rng, seed=seed, num_vertices=4)
-    vertices = P.vertices
-    return Tetrahedron(vertices)
+    return Tetrahedron(P.vertices)
 end