Temp fix for the race condition during the tests. (#7021)

michaelgsharp · web-flow · commit 0d2fd603a20b · 2024-02-21T13:14:45.000-07:00
* temp fixing the race condition during the tests

* making db tests sequential
diff --git a/test/Microsoft.ML.Tests/DatabaseLoaderTests.cs b/test/Microsoft.ML.Tests/DatabaseLoaderTests.cs
@@ -20,6 +20,9 @@
 
 namespace Microsoft.ML.Tests
 {
+    [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]
+    public class NoParallelizationDefinition { }
+    [Collection(nameof(NoParallelizationDefinition))]
     public class DatabaseLoaderTests : BaseTestClass
     {
         public DatabaseLoaderTests(ITestOutputHelper output)
diff --git a/test/Microsoft.ML.Tests/Transformers/WordEmbeddingsTests.cs b/test/Microsoft.ML.Tests/Transformers/WordEmbeddingsTests.cs
@@ -12,6 +12,9 @@
 
 namespace Microsoft.ML.Tests.Transformers
 {
+    [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]
+    public class NoParallelizationDefinition { }
+    [Collection(nameof(NoParallelizationDefinition))]
     public sealed class WordEmbeddingsTests : TestDataPipeBase
     {
         public WordEmbeddingsTests(ITestOutputHelper helper)

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,9 @@`
`20`	`20`
`21`	`21`	`namespace Microsoft.ML.Tests`
`22`	`22`	`{`
	`23`	`+ [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]`
	`24`	`+ public class NoParallelizationDefinition { }`
	`25`	`+ [Collection(nameof(NoParallelizationDefinition))]`
`23`	`26`	`public class DatabaseLoaderTests : BaseTestClass`
`24`	`27`	`{`
`25`	`28`	`public DatabaseLoaderTests(ITestOutputHelper output)`
Original file line number	Diff line number	Diff line change
`@@ -12,6 +12,9 @@`
`12`	`12`
`13`	`13`	`namespace Microsoft.ML.Tests.Transformers`
`14`	`14`	`{`
	`15`	`+ [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]`
	`16`	`+ public class NoParallelizationDefinition { }`
	`17`	`+ [Collection(nameof(NoParallelizationDefinition))]`
`15`	`18`	`public sealed class WordEmbeddingsTests : TestDataPipeBase`
`16`	`19`	`{`
`17`	`20`	`public WordEmbeddingsTests(ITestOutputHelper helper)`