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"