Spaces:
Runtime error
Runtime error
File size: 2,334 Bytes
7288748 |
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 |
import pytest
from youtube_transcriber.video import YoutubeVideo
def test_youtube_video_init():
video = YoutubeVideo(channel_name="The verge",
url="https://www.youtube.com/watch?v=Jzl0hHTc7Jw",
title="Pixel 7 Pro and 7 hands-on: more of the same",
description="Google’s Pixel 7 and 7 Pro...",
transcription=" Seven years ago, we set out...",
segments=[{"start": 0.0, "end": 1.3, "text": " Seven years ago"},
{"start": 1.3, "end": 2.3, "text": " we set out..."}])
assert type(video) == YoutubeVideo
assert video.channel_name == "The verge"
assert video.url == "https://www.youtube.com/watch?v=Jzl0hHTc7Jw"
assert video.title == "Pixel 7 Pro and 7 hands-on: more of the same"
assert video.description == "Google’s Pixel 7 and 7 Pro..."
assert video.transcription == " Seven years ago, we set out..."
assert video.segments == [{"start": 0.0, "end": 1.3, "text": " Seven years ago"},
{"start": 1.3, "end": 2.3, "text": " we set out..."}]
def test_youtube_video_to_tuple():
video = YoutubeVideo(channel_name="The verge",
url="https://www.youtube.com/watch?v=Jzl0hHTc7Jw",
title="Pixel 7 Pro and 7 hands-on: more of the same",
description="Google’s Pixel 7 and 7 Pro...",
transcription=" Seven years ago, we set out...",
segments=[{"start": 0.0, "end": 1.3, "text": " Seven years ago"},
{"start": 1.3, "end": 2.3, "text": " we set out..."}])
video_tuple = video.to_tuple()
assert len(video_tuple) == 6
assert type(video_tuple) == tuple
assert video_tuple[0] == "The verge"
assert video_tuple[1] == "https://www.youtube.com/watch?v=Jzl0hHTc7Jw"
assert video_tuple[2] == "Pixel 7 Pro and 7 hands-on: more of the same"
assert video_tuple[3] == "Google’s Pixel 7 and 7 Pro..."
assert video_tuple[4] == " Seven years ago, we set out..."
assert video_tuple[5] == [{"start": 0.0, "end": 1.3, "text": " Seven years ago"},
{"start": 1.3, "end": 2.3, "text": " we set out..."}] |