diff --git a/.gitattributes b/.gitattributes
deleted file mode 100644
index 5a98245..0000000
--- a/.gitattributes
+++ /dev/null
@@ -1 +0,0 @@
-tests/data/revlog_history.json filter=lfs diff=lfs merge=lfs -text
diff --git a/src/convertor.rs b/src/convertor.rs
index 6d60d0c..8902622 100644
--- a/src/convertor.rs
+++ b/src/convertor.rs
@@ -231,9 +231,9 @@ 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))
@@ -241,10 +241,13 @@ fn test() {
 
     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);
 }
diff --git a/src/dataset.rs b/src/dataset.rs
index 9fd1793..6324b0d 100644
--- a/src/dataset.rs
+++ b/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;
diff --git a/tests/data/revlog_history.json b/tests/data/revlog_history.json
deleted file mode 100644
index 0dffe79..0000000
--- a/tests/data/revlog_history.json
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f74895df09f9d244825afc54d6020a8221af2a19626a3650b3263d10ba63c427
-size 67945