coremlit 0.1.0

Safe, synchronous CoreML runtime for macOS (CPU/GPU/Neural Engine) with opt-in on-device multimodal pipelines: speech (Whisper STT, forced alignment, speaker diarization, Silero VAD), AudioSet sound-event tagging, and audio/text/image embeddings (CLAP, granite, SigLIP)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use super::*;

#[test]
fn contract_mismatch_display_names_feature() {
  let e = Error::ContractMismatch(ContractMismatch::new(
    "input_features",
    "[1, 1, 1001, 64] float32".to_string(),
    "[1, 1, 1001, 64] float16".to_string(),
  ));
  let msg = e.to_string();
  assert!(msg.contains("input_features"), "{msg}");
  assert!(msg.contains("float16"), "{msg}");
}

#[test]
fn output_shape_display_shows_both() {
  let e = Error::OutputShape(OutputShape::new(vec![512, 1], vec![1, 512]));
  let msg = e.to_string();
  assert!(
    msg.contains("[512, 1]") && msg.contains("[1, 512]"),
    "{msg}"
  );
}

#[test]
fn coremlit_errors_convert_via_from() {
  // `#[from]` lets `?` lift coremlit errors into clapkit's Error.
  let e = Error::from(crate::PredictionError::MissingOutput(
    "audio_embeds".to_string(),
  ));
  assert!(matches!(e, Error::Prediction(_)), "got {e:?}");
}

#[test]
fn non_finite_variants_carry_index() {
  assert!(Error::NonFiniteInput(7).to_string().contains('7'));
  assert!(Error::NonFiniteEmbedding(3).to_string().contains('3'));
}

#[test]
fn from_winditerror_is_total_and_does_not_special_case_empty() {
  // The blanket `From<WinditError> for Error` must be a lossless, total wrap
  // into `Error::Windowing` — no variant gets special-cased here. This is the
  // conversion a downstream `SmoothPolicy`/`Smoother` caller takes on a plain
  // `?` (both re-exported at `clap::{SmoothPolicy, Smoother}`), so a custom
  // policy's `WinditError::Empty` on a NONEMPTY stream must surface as
  // `Windowing(Empty)`, not the misleading `EmptyWindows` ("cannot aggregate
  // zero window embeddings"). Only `aggregate()` may map `Empty ->
  // EmptyWindows`, and only at its own call site, where an empty error and an
  // empty input slice are actually the same event.
  let e = Error::from(WinditError::Empty);
  assert!(
    matches!(e, Error::Windowing(WinditError::Empty)),
    "the blanket From<WinditError> impl must not special-case Empty; got {e:?}"
  );
}

#[test]
fn non_finite_output_display_carries_index() {
  let msg = Error::NonFiniteOutput(7).to_string();
  assert!(msg.contains('7'), "{msg}");
  assert!(msg.contains("non-finite"), "{msg}");
}

#[test]
fn embedding_dim_mismatch_display_shows_expected_then_got() {
  let msg = Error::EmbeddingDimMismatch(EmbeddingDimMismatch::new(512, 256)).to_string();
  assert_eq!(msg, "embedding dimension mismatch: expected 512, got 256");
}

#[test]
fn embedding_not_unit_norm_display_carries_the_deviation() {
  let msg = Error::EmbeddingNotUnitNorm(0.25).to_string();
  assert!(msg.contains("unit-norm"), "{msg}");
  assert!(msg.contains("0.25"), "{msg}");
}

#[test]
fn special_token_overhead_display_names_both_numbers() {
  let msg = Error::SpecialTokenOverhead(SpecialTokenOverhead::new(513, 512)).to_string();
  assert!(msg.contains("513"), "{msg}");
  assert!(msg.contains("512"), "{msg}");
  assert!(msg.contains("special tokens"), "{msg}");
}