Fix pytest tests to exit with proper exit code (#16681)

test/srt/test_evs.py

@@ -51,4 +51,6 @@ def test_replace_offsets_with_tokens_per_frame():
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    import sys
+
+    sys.exit(pytest.main([__file__]))

test/srt/test_video_utils.py

@@ -56,4 +56,6 @@ def test_sample_video_frames_lengths(case: Case):
     assert result == case.expected_frames
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    import sys
+
+    sys.exit(pytest.main([__file__]))