Skip to content

Commit 3aa3d27

Browse files
committed
minor: adding partial tests
1 parent 00ad4e9 commit 3aa3d27

1 file changed

Lines changed: 16 additions & 1 deletion

File tree

src/test/java/com/thealgorithms/machinelearning/MultinomialNaiveBayesClassifierTest.java

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,14 @@ class MultinomialNaiveBayesClassifierTest {
1111
@Test
1212
void predictsCorrectClassOnSeparableToyDataset() {
1313
// Class 0 samples are dominated by feature 0; class 1 samples by feature 1.
14-
double[][] features = {{5, 1},{6, 0},{4, 1},{1, 5},{0, 6},{1, 4},};
14+
double[][] features = {
15+
{5, 1},
16+
{6, 0},
17+
{4, 1},
18+
{1, 5},
19+
{0, 6},
20+
{1, 4},
21+
};
1522
int[] labels = {0, 0, 0, 1, 1, 1};
1623

1724
MultinomialNaiveBayesClassifier classifier = new MultinomialNaiveBayesClassifier();
@@ -98,4 +105,12 @@ void mismatchedFeatureAndLabelLengthsThrowsIllegalArgumentException() {
98105
MultinomialNaiveBayesClassifier classifier = new MultinomialNaiveBayesClassifier();
99106
assertThrows(IllegalArgumentException.class, () -> classifier.fit(features, labels));
100107
}
108+
109+
@Test
110+
void emptyFeaturesArrayThrowsIllegalArgumentException() {
111+
double[][] features = {};
112+
int[] labels = {};
113+
MultinomialNaiveBayesClassifier classifier = new MultinomialNaiveBayesClassifier();
114+
assertThrows(IllegalArgumentException.class, () -> classifier.fit(features, labels));
115+
}
101116
}

0 commit comments

Comments
 (0)