test/test_video.py

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..."}]