From 1735ea61df23f342c995fd207dfbbdb7182ae164 Mon Sep 17 00:00:00 2001
From: Volodymyr Orlov <vorl@Echo.local>
Date: Fri, 4 Sep 2020 10:10:41 -0700
Subject: [PATCH] fix: broken unit tests

---
 src/ensemble/random_forest_classifier.rs | 4 +++-
 src/linalg/ndarray_bindings.rs           | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/ensemble/random_forest_classifier.rs b/src/ensemble/random_forest_classifier.rs
index 418f583..77352a3 100644
--- a/src/ensemble/random_forest_classifier.rs
+++ b/src/ensemble/random_forest_classifier.rs
@@ -222,6 +222,7 @@ impl<T: RealNumber> RandomForestClassifier<T> {
 mod tests {
     use super::*;
     use crate::linalg::naive::dense_matrix::DenseMatrix;
+    use crate::metrics::*;
 
     #[test]
     fn fit_predict_iris() {
@@ -264,7 +265,8 @@ mod tests {
             },
         );
 
-        assert_eq!(y, classifier.predict(&x));
+        assert!(accuracy(&y, &classifier.predict(&x)) > 0.9);
+        
     }
 
     #[test]
diff --git a/src/linalg/ndarray_bindings.rs b/src/linalg/ndarray_bindings.rs
index 122673a..7a2a740 100644
--- a/src/linalg/ndarray_bindings.rs
+++ b/src/linalg/ndarray_bindings.rs
@@ -735,7 +735,7 @@ mod tests {
                 min_samples_leaf: 1,
                 min_samples_split: 2,
                 n_trees: 1000,
-                mtry: Option::None,
+                m: Option::None,
             },
         )
         .predict(&x);