Skip to content

Commit 36fcaee

Browse files
committed
Update the Unit test for KNN
1 parent 6b77caa commit 36fcaee

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

Algorithms.Tests/MachineLearning/KNearestNeighborsTests.cs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,8 @@ public void EuclideanDistance_CorrectResult()
5757
public void Predict_SingleNeighbor_CorrectLabel()
5858
{
5959
var knn = new KNearestNeighbors<string>(1);
60-
knn.AddSample(new double[] { 1.0, 2.0 }, "A");
61-
knn.AddSample(new double[] { 3.0, 4.0 }, "B");
60+
knn.AddSample(new[] { 1.0, 2.0 }, "A");
61+
knn.AddSample(new[] { 3.0, 4.0 }, "B");
6262
var label = knn.Predict(new[] { 1.1, 2.1 });
6363
Assert.That(label, Is.EqualTo("A"));
6464
}
@@ -67,9 +67,9 @@ public void Predict_SingleNeighbor_CorrectLabel()
6767
public void Predict_MajorityVote_CorrectLabel()
6868
{
6969
var knn = new KNearestNeighbors<string>(3);
70-
knn.AddSample(new double[] { 0.0, 0.0 }, "A");
71-
knn.AddSample(new double[] { 0.1, 0.1 }, "A");
72-
knn.AddSample(new double[] { 1.0, 1.0 }, "B");
70+
knn.AddSample(new[] { 0.0, 0.0 }, "A");
71+
knn.AddSample(new[] { 0.1, 0.1 }, "A");
72+
knn.AddSample(new[] { 1.0, 1.0 }, "B");
7373
var label = knn.Predict(new[] { 0.05, 0.05 });
7474
Assert.That(label, Is.EqualTo("A"));
7575
}
@@ -78,8 +78,8 @@ public void Predict_MajorityVote_CorrectLabel()
7878
public void Predict_TieBreaker_ReturnsConsistentLabel()
7979
{
8080
var knn = new KNearestNeighbors<string>(2);
81-
knn.AddSample(new double[] { 0.0, 0.0 }, "A");
82-
knn.AddSample(new double[] { 1.0, 1.0 }, "B");
81+
knn.AddSample(new[] { 0.0, 0.0 }, "A");
82+
knn.AddSample(new[] { 1.0, 1.0 }, "B");
8383
var label = knn.Predict(new[] { 0.5, 0.5 });
8484
Assert.That(label, Is.EqualTo("A"));
8585
}

0 commit comments

Comments
 (0)