Skip to content

Commit b30b013

Browse files
Copilotcrossjam
andcommitted
Fix formatting and verify pydantic model implementation
Co-authored-by: crossjam <208062+crossjam@users.noreply.github.com>
1 parent 1bc86d5 commit b30b013

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

tests/test_transcription_models.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -17,19 +17,15 @@ class TestTranscriptionSegmentModel:
1717

1818
def test_valid_segment(self):
1919
"""Test creating a valid segment model."""
20-
segment = TranscriptionSegmentModel(
21-
start=0.0, end=5.0, text="Hello world", speaker=None
22-
)
20+
segment = TranscriptionSegmentModel(start=0.0, end=5.0, text="Hello world", speaker=None)
2321
assert segment.start == 0.0
2422
assert segment.end == 5.0
2523
assert segment.text == "Hello world"
2624
assert segment.speaker is None
2725

2826
def test_segment_with_speaker(self):
2927
"""Test segment with speaker diarization."""
30-
segment = TranscriptionSegmentModel(
31-
start=0.0, end=5.0, text="Hello", speaker="SPEAKER_1"
32-
)
28+
segment = TranscriptionSegmentModel(start=0.0, end=5.0, text="Hello", speaker="SPEAKER_1")
3329
assert segment.speaker == "SPEAKER_1"
3430

3531
def test_negative_start_time(self):
@@ -52,9 +48,7 @@ def test_missing_required_fields(self):
5248
def test_extra_fields_forbidden(self):
5349
"""Test that extra fields are forbidden."""
5450
with pytest.raises(ValidationError) as exc_info:
55-
TranscriptionSegmentModel(
56-
start=0.0, end=5.0, text="Test", extra_field="not allowed"
57-
)
51+
TranscriptionSegmentModel(start=0.0, end=5.0, text="Test", extra_field="not allowed")
5852
assert "extra_field" in str(exc_info.value)
5953

6054
def test_json_serialization(self):

0 commit comments

Comments
 (0)