@@ -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