Committing before copilot touches this

breadpad-shared/tests/classifier.rs

@@ -0,0 +1,83 @@
+use breadpad_shared::classifier::Classifier;
+use breadpad_shared::types::NoteType;
+use chrono::Timelike;
+
+fn cl() -> Classifier {
+    Classifier::load("auto", "08:00")
+}
+
+#[test]
+fn active_provider_is_cpu() {
+    // QNN and Vulkan EPs are not compiled in; CPU is always the fallback.
+    let c = cl();
+    assert_eq!(c.active_provider, breadpad_shared::classifier::ExecutionProvider::Cpu);
+}
+
+#[test]
+fn classify_falls_back_to_rule_based() {
+    let mut c = cl();
+    let r = c.classify("buy milk");
+    assert_eq!(r.note_type, NoteType::Todo);
+    assert!(r.time.is_none());
+}
+
+#[test]
+fn classify_todo_via_fallback() {
+    let mut c = cl();
+    assert_eq!(c.classify("fix the segfault").note_type, NoteType::Todo);
+}
+
+#[test]
+fn classify_reminder_via_fallback() {
+    let mut c = cl();
+    let r = c.classify("call mum at 6pm");
+    assert_eq!(r.note_type, NoteType::Reminder);
+    assert!(r.time.is_some());
+}
+
+#[test]
+fn classify_idea_via_fallback() {
+    let mut c = cl();
+    assert_eq!(c.classify("what if we added a calendar view").note_type, NoteType::Idea);
+}
+
+#[test]
+fn classify_question_via_fallback() {
+    let mut c = cl();
+    assert_eq!(c.classify("why does this fail?").note_type, NoteType::Question);
+}
+
+#[test]
+fn classify_note_via_fallback() {
+    let mut c = cl();
+    assert_eq!(c.classify("meeting went well today").note_type, NoteType::Note);
+}
+
+#[test]
+fn classify_recurrence_via_fallback() {
+    let mut c = cl();
+    let r = c.classify("standup every monday at 9am");
+    assert!(r.rrule.is_some(), "expected rrule from fallback parser");
+    assert_eq!(r.note_type, NoteType::Reminder);
+}
+
+#[test]
+fn classify_custom_morning_time() {
+    let mut c = Classifier::load("auto", "07:15");
+    let r = c.classify("sync tomorrow morning");
+    let t = r.time.expect("should have a time for tomorrow morning");
+    let local: chrono::DateTime<chrono::Local> = t.into();
+    assert_eq!(local.hour(), 7);
+    assert_eq!(local.minute(), 15);
+}
+
+#[test]
+fn model_path_points_to_expected_location() {
+    let c = cl();
+    assert!(
+        c.model_path.to_str().unwrap().contains("breadpad"),
+        "model path: {:?}",
+        c.model_path
+    );
+    assert!(c.model_path.to_str().unwrap().ends_with("classifier.onnx"));
+}