fixes

Author: CarloLucibello

GNNGraphs/test/datastore.jl

@@ -38,7 +38,7 @@ end
     @test ds2.x == ds.x .+ 1
     @test ds2.y == ds.y .+ 1
 
-    @test_throws AssertionError ds2=map(x -> [x; x], ds)
+    @test_throws AssertionError ds2 == map(x -> [x; x], ds)
 end
 
 @testitem "getdata / getn" begin

GNNGraphs/test/utils.jl

@@ -94,15 +94,16 @@
 end
 
 @testitem "color_refinement" setup=[GraphsTestModule] begin
+    # TODO: this passes only on julia 1.10. Use a deterministic graph generator
     using .GraphsTestModule
     for GRAPH_T in GRAPH_TYPES
         rng = MersenneTwister(17)
         g = rand_graph(rng, 10, 20, graph_type = GRAPH_T)
         x0 = ones(Int, 10)
         x, ncolors, niters = color_refinement(g, x0)
-        @test ncolors == 8
+        @test ncolors == 7
         @test niters == 2
-        @test x == [6, 7, 8, 9, 10, 11, 9, 11, 12, 13]
+        @test x == [1, 1, 6, 6, 6, 7, 8, 9, 10, 11] 
 
         x2, _, _ = color_refinement(g)
         @test x2 == x