fix: broken tests

This commit is contained in:
Volodymyr Orlov
2020-06-05 18:10:23 -07:00
parent 45cf7adf37
commit 3982ae8f10
2 changed files with 3 additions and 4 deletions
+2 -2
View File
@@ -323,7 +323,7 @@ mod tests {
.approximate_eq(&expected_eigenvectors.abs(), 1e-4)); .approximate_eq(&expected_eigenvectors.abs(), 1e-4));
for i in 0..pca.eigenvalues.len() { for i in 0..pca.eigenvalues.len() {
assert_eq!(pca.eigenvalues[i].abs(), expected_eigenvalues[i].abs()); assert!((pca.eigenvalues[i].abs() - expected_eigenvalues[i].abs()).abs() < 1e-8);
} }
let us_arrests_t = pca.transform(&us_arrests); let us_arrests_t = pca.transform(&us_arrests);
@@ -438,7 +438,7 @@ mod tests {
.approximate_eq(&expected_eigenvectors.abs(), 1e-4)); .approximate_eq(&expected_eigenvectors.abs(), 1e-4));
for i in 0..pca.eigenvalues.len() { for i in 0..pca.eigenvalues.len() {
assert_eq!(pca.eigenvalues[i].abs(), expected_eigenvalues[i].abs()); assert!((pca.eigenvalues[i].abs() - expected_eigenvalues[i].abs()).abs() < 1e-8);
} }
let us_arrests_t = pca.transform(&us_arrests); let us_arrests_t = pca.transform(&us_arrests);
+1 -2
View File
@@ -264,8 +264,7 @@ impl<T: FloatExt, M: Matrix<T>> LogisticRegression<T, M> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use crate::linalg::naive::dense_matrix::*; use crate::linalg::naive::dense_matrix::*;
use crate::metrics::*;
use ndarray::{arr1, arr2, Array1}; use ndarray::{arr1, arr2, Array1};
#[test] #[test]