few more tests

learned.json

@@ -1,8 +1,4 @@
 {
-  "learned_images": {
-    "/////8/wz/DP8Ofwx/OH8wP5A/kH+Z/5//////////8=": "47",
-    "//////////////////////////////////////////8=": ""
-  },
   "learned_tracks": {
     "//////////+f/x//H/7f/N/A34HPn8+DD8AfwP////8=": "Magdalena Club",
     "/////z/wv/+/zz/vf+Z/5P/IH9gP2c+cD8CfwP////8=": "Rennvoort",

src/configs/learned.default.json

@@ -1,4 +1,3 @@
 {
-    "learned_images": {},
     "learned_tracks": {}
 }

src/ocr_db.rs

@@ -1,7 +1,8 @@
-use std::{collections::HashMap, sync::{RwLock}};
+use std::{collections::HashMap, sync::RwLock};
 
 use crate::{
-    config::{load_config_or_make_default, save_json_config}, ocr,
+    config::{load_config_or_make_default, save_json_config},
+    ocr,
 };
 
 use anyhow::Result;
@@ -60,7 +61,10 @@ impl OcrDatabase {
     }
 
     pub fn learn(&self, hash: &str, val: char) {
-        self.learned_chars.write().unwrap().push((ImageHash::from_base64(hash).unwrap(), val));
+        self.learned_chars
+            .write()
+            .unwrap()
+            .push((ImageHash::from_base64(hash).unwrap(), val));
     }
 
     pub fn learn_phrase(&self, hashes: &[ImageHash], phrase: &str) -> Result<()> {
@@ -83,13 +87,20 @@ impl OcrDatabase {
 
     pub fn ocr_char(&self, hash: &ImageHash) -> Option<char> {
         let state = self.learned_chars.read().unwrap();
-        let (_, c) = state.iter().min_by_key(|(learned_hash, _)| hash.dist(learned_hash))?;
+        let (_, c) = state
+            .iter()
+            .min_by_key(|(learned_hash, _)| hash.dist(learned_hash))?;
         Some(*c)
     }
 
     pub fn ocr_hashes(&self, hashes: &[ImageHash]) -> String {
-        let buffer: String = hashes.iter().filter_map(|hash| self.ocr_char(hash)).collect();
-        buffer.trim_end_matches(|c: char| !c.is_alphanumeric()).to_owned()
+        let buffer: String = hashes
+            .iter()
+            .filter_map(|hash| self.ocr_char(hash))
+            .collect();
+        buffer
+            .trim_end_matches(|c: char| !c.is_alphanumeric())
+            .to_owned()
     }
 
     pub fn ocr_image(&self, image: &RgbImage) -> String {
@@ -97,3 +108,22 @@ impl OcrDatabase {
         self.ocr_hashes(&hashes)
     }
 }
+
+#[test]
+fn test_ocr() {
+    let raw: RawOcrDatabase =
+        serde_json::from_str(include_str!("configs/ocr.default.json")).unwrap();
+    let db: OcrDatabase = (&raw).into();
+
+    let image = image::load_from_memory(include_bytes!("test_data/test-image-3.png")).unwrap().to_rgb8();
+    assert_eq!(db.ocr_image(&image), "00:30.625");
+
+    let image = image::load_from_memory(include_bytes!("test_data/test-image-4.png")).unwrap().to_rgb8();
+    assert_eq!(db.ocr_image(&image), "00:20.296");
+
+    let image = image::load_from_memory(include_bytes!("test_data/test-image-num-1.png")).unwrap().to_rgb8();
+    assert_eq!(db.ocr_image(&image), "1");
+
+    let image = image::load_from_memory(include_bytes!("test_data/test-image-blank.png")).unwrap().to_rgb8();
+    assert_eq!(db.ocr_image(&image), "");
+}