rewrite test for convertor

src/convertor.rs

@@ -231,20 +231,23 @@ pub fn anki_to_fsrs() -> Vec<FSRSItem> {
 #[test]
 fn test() {
     let revlogs = read_collection();
-    dbg!(revlogs.len());
+    assert_eq!(revlogs.len(), 24394);
     let revlogs_per_card = group_by_cid(revlogs);
-    dbg!(revlogs_per_card.len());
+    assert_eq!(revlogs_per_card.len(), 3324);
     let mut extracted_revlogs_per_card: Vec<Vec<RevlogEntry>> = revlogs_per_card
         .into_iter()
         .map(|entries| extract_time_series_feature(entries, 4, Tz::Asia__Shanghai))
         .collect();
 
     dbg!(&extracted_revlogs_per_card[0]);
     extracted_revlogs_per_card = remove_non_learning_first(extracted_revlogs_per_card);
-    dbg!(extracted_revlogs_per_card
-        .iter()
-        .map(|x| x.len())
-        .sum::<usize>());
+    assert_eq!(
+        extracted_revlogs_per_card
+            .iter()
+            .map(|x| x.len())
+            .sum::<usize>(),
+        17614
+    );
     let fsrs_items: Vec<FSRSItem> = convert_to_fsrs_items(extracted_revlogs_per_card);
-    dbg!(fsrs_items.len());
+    assert_eq!(fsrs_items.len(), 14290);
 }

src/dataset.rs

@@ -128,32 +128,6 @@ impl FSRSDataset {
     }
 }
 
-#[test]
-fn test_from_json() {
-    const JSON_FILE: &str = "tests/data/revlog_history.json";
-    use burn::data::dataloader::DataLoaderBuilder;
-    use burn::data::dataloader::Dataset;
-    use burn::data::dataset::InMemDataset;
-    let dataset = InMemDataset::<FSRSItem>::from_json_rows(JSON_FILE).unwrap();
-    let item = dataset.get(704).unwrap();
-    dbg!(&item);
-
-    use burn_ndarray::NdArrayBackend;
-    use burn_ndarray::NdArrayDevice;
-    let device = NdArrayDevice::Cpu;
-    type Backend = NdArrayBackend<f32>;
-    let batcher = FSRSBatcher::<Backend>::new(device);
-    let dataloader = DataLoaderBuilder::new(batcher)
-        .batch_size(1)
-        .shuffle(42)
-        .num_workers(4)
-        .build(dataset);
-    for item in dataloader.iter() {
-        dbg!(&item.r_historys);
-        break;
-    }
-}
-
 #[test]
 fn test_from_anki() {
     use burn::data::dataloader::Dataset;