Skip to content

Commit 75deff4

Browse files
authored
Add KNNTest class for KNN classification testing
1 parent e98990b commit 75deff4

1 file changed

Lines changed: 20 additions & 0 deletions

File tree

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
package com.thealgorithms.machinelearning;
2+
3+
import java.util.ArrayList;
4+
import java.util.List;
5+
import org.junit.jupiter.api.Assertions;
6+
import org.junit.jupiter.api.Test;
7+
8+
public class KNNTest {
9+
@Test
10+
public void testClassifyStandard() {
11+
List<KNN.DataPoint> dataset = new ArrayList<>();
12+
dataset.add(new KNN.DataPoint(new double[] {1.0, 1.0}, "GroupA"));
13+
dataset.add(new KNN.DataPoint(new double[] {1.5, 2.0}, "GroupA"));
14+
dataset.add(new KNN.DataPoint(new double[] {8.0, 9.0}, "GroupB"));
15+
16+
double[] target = new double[] {1.2, 1.3};
17+
String result = KNN.classify(dataset, target, 2);
18+
Assertions.assertEquals("GroupA", result);
19+
}
20+
}

0 commit comments

Comments
 (0)