def test_retrieve_video_length(self): vid_id = yt.retrieve_video_id(self.search_result) assert yt.retrieve_video_length(self.youtube, vid_id) == "sample video length"
def test_retrieve_video_id(self): assert yt.retrieve_video_id(self.search_result) == "sample video ID"