testing light gbm tests sequentially (dotnet#6968)

ibrahim324 · Jan 23, 2024 · 54fa44f · 54fa44f
1 parent 125b6d5
commit 54fa44f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/test/Microsoft.ML.TestFramework/BaseTestClass.cs b/test/Microsoft.ML.TestFramework/BaseTestClass.cs
@@ -81,7 +81,7 @@ void IDisposable.Dispose()
             Cleanup();
             Process proc = Process.GetCurrentProcess();
             Console.WriteLine($"Finished test: {FullTestName} " +
-                $"with memory usage {proc.WorkingSet64.ToString("N", CultureInfo.InvariantCulture)}");
+                $"with memory usage {proc.WorkingSet64.ToString("N", CultureInfo.InvariantCulture)} and max memory usage {proc.PeakWorkingSet64.ToString("N", CultureInfo.InvariantCulture)}");
         }
 
         protected virtual void Initialize()

diff --git a/test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs b/test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs
@@ -24,6 +24,10 @@
 
 namespace Microsoft.ML.Tests.TrainerEstimators
 {
+    [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]
+    public class NoParallelizationDefinition { }
+
+    [Collection(nameof(NoParallelizationDefinition))]
     public partial class TrainerEstimators : TestDataPipeBase
     {
         /// <summary>