Make sure SimpleGraph uses RNG (#1500)

Co-authored-by: oisinfaust <[email protected]>
sbromberger · Dec 13, 2020 · ab65981 · ab65981
1 parent 616e715
commit ab65981
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/SimpleGraphs/generators/randgraphs.jl b/src/SimpleGraphs/generators/randgraphs.jl
@@ -26,7 +26,7 @@ function SimpleGraph{T}(nv::Integer, ne::Integer; seed::Int=-1) where T <: Integ
     tnv = T(nv)
     maxe = div(Int(nv) * (nv - 1), 2)
     @assert(ne <= maxe, "Maximum number of edges for this graph is $maxe")
-    ne > div((2 * maxe), 3)  && return complement(SimpleGraph(tnv, maxe - ne))
+    ne > div((2 * maxe), 3)  && return complement(SimpleGraph(tnv, maxe - ne, seed=seed))
 
     rng = getRNG(seed)
     g = SimpleGraph(tnv)
@@ -63,7 +63,7 @@ function SimpleDiGraph{T}(nv::Integer, ne::Integer; seed::Int=-1) where T <: Int
     tnv = T(nv)
     maxe = Int(nv) * (nv - 1)
     @assert(ne <= maxe, "Maximum number of edges for this graph is $maxe")
-    ne > div((2 * maxe), 3) && return complement(SimpleDiGraph{T}(tnv, maxe - ne))
+    ne > div((2 * maxe), 3) && return complement(SimpleDiGraph{T}(tnv, maxe - ne, seed=seed))
 
     rng = getRNG(seed)
     g = SimpleDiGraph(tnv)

diff --git a/test/simplegraphs/generators/randgraphs.jl b/test/simplegraphs/generators/randgraphs.jl
@@ -10,7 +10,9 @@
         @test eltype(r2) == Int
 
         @test SimpleGraph(10, 20, seed=3) == SimpleGraph(10, 20, seed=3)
+        @test SimpleGraph(10, 40, seed=3) == SimpleGraph(10, 40, seed=3)
         @test SimpleDiGraph(10, 20, seed=3) == SimpleDiGraph(10, 20, seed=3)
+        @test SimpleDiGraph(10, 80, seed=3) == SimpleDiGraph(10, 80, seed=3)
         @test SimpleGraph(10, 20, seed=3) == erdos_renyi(10, 20, seed=3)
         @test ne(Graph(10, 40, seed=3)) == 40
         @test ne(DiGraph(10, 80, seed=3)) == 80