From b05b84832d6c713f51523fbed1faee861b179b09 Mon Sep 17 00:00:00 2001
From: Xuanzhao Gao <gaoxuanzhao@gmail.com>
Date: Tue, 28 May 2024 21:27:31 +0800
Subject: [PATCH] update tests

---
 src/greedy.jl  | 4 ++--
 test/greedy.jl | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/greedy.jl b/src/greedy.jl
index 2d000e9..683d029 100644
--- a/src/greedy.jl
+++ b/src/greedy.jl
@@ -3,14 +3,14 @@ struct ContractionTree
     right
 end
 
-```
+"""
     Greedy{TA, TT}
     * `α` is the parameter for the loss function, for pairwise interaction, L = size(out) - α * (size(in1) + size(in2))
     * `tempareture` is the parameter for sampling, if it is zero, the minimum loss is selected; for non-zero, the loss is selected by the Boltzmann distribution, given by p ~ exp(-loss/tempareture).
 
     MinSpaceOut() = Greedy(0.0, 0.0)
     MinSpaceDiff() = Greedy(1.0, 0.0)
-```
+"""
 struct Greedy{TA, TT}
     α::TA
     tempareture::TT
diff --git a/test/greedy.jl b/test/greedy.jl
index ae0f9b1..1f33122 100644
--- a/test/greedy.jl
+++ b/test/greedy.jl
@@ -60,6 +60,7 @@ end
 end
 
 @testset "fullerene" begin
+    Random.seed!(123)
     function fullerene()
         φ = (1+√5)/2
         res = NTuple{3,Float64}[]
@@ -112,8 +113,8 @@ end
     @test flatten(optcode) == code
     @test flatten(code) == code
 
-    optcode_hyper = optimize_greedy(code, size_dict, method = Greedy(0.0, 100.0))
+    optcode_hyper = optimize_greedy(code, size_dict, method = Greedy(0.0, 100.0), nrepeat = 20)
     cc3 = contraction_complexity(optcode_hyper, edge_sizes)
-    @test cc3.sc == 10
+    @test cc3.sc <= 12
     @test flatten(optcode_hyper) == code
 end
\ No newline at end of file