Add OptunaTrial and Pruner to ProgressState in tests

hrntsm · Dec 22, 2024 · 2d598d7 · 2d598d7
1 parent 25d2c35
commit 2d598d7
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/Tunny.CoreTests/Handler/ProgressStateTests.cs b/Tunny.CoreTests/Handler/ProgressStateTests.cs
@@ -1,5 +1,10 @@
 using System.Collections.Generic;
 
+using Optuna.Trial;
+
+using Tunny.Core.Settings;
+using Tunny.Core.TEnum;
+
 using Xunit;
 
 namespace Tunny.Core.Handler.Tests
@@ -26,7 +31,10 @@ public void ProgressStatePropertiesTest()
                 BestValues = bests,
                 HypervolumeRatio = 0,
                 EstimatedTimeRemaining = new System.TimeSpan(),
-                IsReportOnly = false
+                IsReportOnly = false,
+                OptunaTrial = new TrialWrapper(1),
+                Pruner = new Pruner(),
+                PercentComplete = 0
             };
 
             Assert.Equal(param, progressState.Parameter);
@@ -36,6 +44,9 @@ public void ProgressStatePropertiesTest()
             Assert.Equal(0, progressState.HypervolumeRatio);
             Assert.Equal(new System.TimeSpan(), progressState.EstimatedTimeRemaining);
             Assert.False(progressState.IsReportOnly);
+            Assert.Equal(1, progressState.OptunaTrial.PyObject);
+            Assert.Equal(PrunerType.None, progressState.Pruner.Type);
+            Assert.Equal(0, progressState.PercentComplete);
         }
 
         [Fact]