Fix code coverage

viceroypenguin · Nov 26, 2023 · a3c112c · a3c112c
1 parent 172dbf8
commit a3c112c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/Source/SuperLinq/Permutations.cs b/Source/SuperLinq/Permutations.cs
@@ -41,8 +41,7 @@ public static IEnumerable<IList<T>> Permutations<T>(this IEnumerable<T> sequence
 	{
 		ArgumentNullException.ThrowIfNull(sequence);
 
-		if (sequence.TryGetCollectionCount() is int count
-			&& count > 21)
+		if (sequence.TryGetCollectionCount() is int and > 21)
 		{
 			ThrowHelper.ThrowArgumentException(nameof(sequence), "Input set is too large to permute properly.");
 		}

diff --git a/Tests/SuperLinq.Test/PermutationsTest.cs b/Tests/SuperLinq.Test/PermutationsTest.cs
@@ -22,15 +22,23 @@ public void TestCardinalityZeroPermutation()
 		permutations.Single().AssertSequenceEqual();
 	}
 
-	[Fact]
-	public void TestExceptionWhenTooLong()
+	public static IEnumerable<object[]> GetTooLongSequences() =>
+		Enumerable.Range(1, 22)
+			.GetBreakingCollectionSequences()
+			.Select(x => new object[] { x, });
+
+	[Theory]
+	[MemberData(nameof(GetTooLongSequences))]
+	public void TestExceptionWhenTooLong(IDisposableEnumerable<int> seq)
 	{
-		using var emptySet = Enumerable.Range(1, 22).AsTestingSequence();
-
-		var result = emptySet.Permutations();
+		using (seq)
+		{
+			var ex = Assert.Throws<ArgumentException>(
+				"sequence",
+				() => seq.Permutations().Consume());
 
-		var ex = Assert.Throws<ArgumentException>("sequence", result.Consume);
-		Assert.Equal("Input set is too large to permute properly. (Parameter 'sequence')", ex.Message);
+			Assert.Equal("Input set is too large to permute properly. (Parameter 'sequence')", ex.Message);
+		}
 	}
 
 	/// <summary>