fix tests

smartcorelib · Mar 4, 2024 · 92cb697 · 92cb697
1 parent 32d7e45
commit 92cb697
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/naive_bayes/mod.rs b/src/naive_bayes/mod.rs
@@ -169,7 +169,7 @@ mod tests {
 
     #[test]
     fn test_predict() {
-        let matrix = DenseMatrix::from_2d_array(&[&[1, 2, 3], &[4, 5, 6], &[7, 8, 9]]);
+        let matrix = DenseMatrix::from_2d_array(&[&[1, 2, 3], &[4, 5, 6], &[7, 8, 9]]).unwrap();
 
         let val = vec![];
         match Model::fit(TestDistribution(&val)).unwrap().predict(&matrix) {

diff --git a/src/tree/decision_tree_classifier.rs b/src/tree/decision_tree_classifier.rs
@@ -1067,7 +1067,7 @@ mod tests {
             &[0., 0., 1., 1.],
             &[0., 0., 0., 0.],
             &[0., 0., 0., 1.],
-        ]);
+        ]).unwrap();
         let y: Vec<u32> = vec![1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0];
         let tree = DecisionTreeClassifier::fit(&x, &y, Default::default()).unwrap();
         assert_eq!(