assert length in test_filter_outlier

open-spaced-repetition · Nov 20, 2023 · dc9b2fe · dc9b2fe
1 parent 88d0794
commit dc9b2fe
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/dataset.rs b/src/dataset.rs
@@ -424,10 +424,10 @@ mod tests {
         let (mut pretrainset, mut trainset): (Vec<FSRSItem>, Vec<FSRSItem>) = dataset
             .into_iter()
             .partition(|item| item.reviews.len() == 2);
-        dbg!(pretrainset.len());
-        dbg!(trainset.len());
+        assert_eq!(pretrainset.len(), 3315);
+        assert_eq!(trainset.len(), 10806);
         (pretrainset, trainset) = filter_outlier(pretrainset, trainset);
-        dbg!(pretrainset.len());
-        dbg!(trainset.len());
+        assert_eq!(pretrainset.len(), 3265);
+        assert_eq!(trainset.len(), 10731);
     }
 }