chore: adiciona .gitignore e commit.command

code/tests/test_diarize.py

@@ -0,0 +1,86 @@
+"""Tests for fcpxml/diarize.py — speaker diarization (WHISPERX-inspired).
+
+The pure assignment helpers are fully testable without pyannote; the
+diarize()/diarization_capability() integration path degrades gracefully when
+pyannote or a token is absent.
+"""
+
+
+from fcpxml.diarize import (
+    DEFAULT_SPEAKER,
+    assign_speakers,
+    build_speakers,
+    diarization_capability,
+    diarize,
+)
+
+
+def test_capability_without_token_is_false():
+    ok, _msg = diarization_capability("")
+    assert ok is False
+
+
+def test_capability_without_pyannote_is_false(monkeypatch):
+    import builtins
+
+    real_import = builtins.__import__
+
+    def block(name, *args, **kwargs):
+        if name.startswith("pyannote"):
+            raise ImportError("blocked for test")
+        return real_import(name, *args, **kwargs)
+
+    monkeypatch.setattr(builtins, "__import__", block)
+    ok, msg = diarization_capability("hf_xyz")
+    assert ok is False
+    assert "pyannote" in msg.lower()
+
+
+def test_diarize_without_token_returns_none():
+    assert diarize("ignored.wav", None) is None
+    assert diarize("ignored.wav", "") is None
+
+
+def test_assign_speakers_no_tracks_uses_default():
+    segments = [{"start": 0.0, "end": 1.0}, {"start": 2.0, "end": 3.0}]
+    words = [{"start": 0.2, "end": 0.4}]
+    segs, ws = assign_speakers(segments, words, None)
+    assert all(s["speaker_id"] == DEFAULT_SPEAKER for s in segs)
+    assert ws[0]["speaker_id"] == DEFAULT_SPEAKER
+
+
+def test_assign_speakers_overlap_mapping():
+    tracks = [(0.0, 2.0, "SPEAKER_A"), (2.0, 4.0, "SPEAKER_B")]
+    segments = [{"start": 0.5, "end": 1.5}, {"start": 2.5, "end": 3.5}]
+    words = [{"start": 0.9, "end": 1.0}, {"start": 3.0, "end": 3.1}]
+    segs, ws = assign_speakers(segments, words, tracks)
+    assert segs[0]["speaker_id"] == "SPEAKER_00"
+    assert segs[1]["speaker_id"] == "SPEAKER_01"
+    assert ws[0]["speaker_id"] == "SPEAKER_00"
+    assert ws[1]["speaker_id"] == "SPEAKER_01"
+
+
+def test_assign_speakers_does_not_mutate_input():
+    tracks = [(0.0, 2.0, "SPEAKER_A")]
+    segments = [{"start": 0.5, "end": 1.5}]
+    words = [{"start": 0.9, "end": 1.0}]
+    assign_speakers(segments, words, tracks)
+    assert "speaker_id" not in segments[0]
+    assert "speaker_id" not in words[0]
+
+
+def test_build_speakers_dedupes_and_orders():
+    segments = [
+        {"speaker_id": "SPEAKER_01"},
+        {"speaker_id": "SPEAKER_00"},
+        {"speaker_id": "SPEAKER_01"},
+    ]
+    speakers = build_speakers(segments)
+    assert speakers == [
+        {"id": "SPEAKER_01", "name": "Speaker 1"},
+        {"id": "SPEAKER_00", "name": "Speaker 2"},
+    ]
+
+
+def test_build_speakers_empty():
+    assert build_speakers([]) == []