ffaf07d628
* [FEATURE] Add support to scrape chapters from comments * tested for comments * remove timestamps file * change order of docstrings
74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
import pytest
|
|
|
|
from ytdl_sub.utils.chapters import Chapters
|
|
from ytdl_sub.utils.chapters import Timestamp
|
|
|
|
|
|
class TestTimestamp:
|
|
@pytest.mark.parametrize(
|
|
"timestamp_str, timestamp_int",
|
|
[
|
|
("0:00", 0),
|
|
("0:24", 24),
|
|
("1:11", 71),
|
|
("01:11", 71),
|
|
("00:22", 22),
|
|
("1:01:01", 3600 + 60 + 1),
|
|
("01:01:01", 3600 + 60 + 1),
|
|
("00:00:00", 0),
|
|
],
|
|
)
|
|
def test_timestamp_from_str(self, timestamp_str, timestamp_int):
|
|
ts = Timestamp.from_str(timestamp_str=timestamp_str)
|
|
assert ts.timestamp_sec == timestamp_int
|
|
|
|
|
|
@pytest.fixture
|
|
def chapter_description_1() -> str:
|
|
return """Support the artist by purchasing the record here:
|
|
https://levanika.bandcamp.com/album/p...
|
|
|
|
Album Art by: po
|
|
|
|
Tracklist:
|
|
00:00 intro
|
|
00:58 audioclip p94568
|
|
01:18 saskipao
|
|
03:32 fifqeby
|
|
05:40 dream
|
|
07:22 rulji
|
|
08:54 modymody
|
|
11:23 .ishos
|
|
16:09 pos
|
|
18:47 lamazybalaxy
|
|
|
|
Denivarlevy Socials:
|
|
|
|
https://open.spotify.com/artist/2XdIT...
|
|
https://youtube.com/channel/UCgI_vAC3...
|
|
"""
|
|
|
|
|
|
@pytest.fixture
|
|
def chapter_description_2() -> str:
|
|
return """01. 00:00 Ocean
|
|
02. 02:41 Dreams
|
|
03. 05:16 Future Tales
|
|
04. 08:50 Mind Travelling
|
|
05. 11:05 Love Supreme
|
|
06. 14:17 Reflections
|
|
07. 16:32 Moonlight Fading
|
|
08. 19:40 Between Two Worlds
|
|
"""
|
|
|
|
|
|
class TestChapters:
|
|
def test_chapters_from_str_1(self, chapter_description_1):
|
|
chapters = Chapters.from_string(chapter_description_1)
|
|
assert len(chapters) == 10
|
|
assert chapters.timestamps[-1].readable_str == "18:47"
|
|
|
|
def test_chapters_from_str_2(self, chapter_description_2):
|
|
chapters = Chapters.from_string(chapter_description_2)
|
|
assert len(chapters) == 8
|
|
assert chapters.timestamps[1].readable_str == "2:41"
|