Skip to content

Commit 5729632

Browse files
committed
refactor: inline fetch_title calls in transcript test cases
1 parent 09aac3e commit 5729632

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/test_mcp.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,8 @@ async def test_list_tools(mcp_client_session: ClientSession) -> None:
4646
async def test_get_transcript(mcp_client_session: ClientSession) -> None:
4747
video_id = "LPZh9BOjkQs"
4848

49-
title = fetch_title(video_id, "en")
5049
expect = Transcript(
51-
title=title,
50+
title=fetch_title(video_id, "en"),
5251
transcript="\n".join((item.text for item in YouTubeTranscriptApi().fetch(video_id))),
5352
)
5453

@@ -70,9 +69,8 @@ async def test_get_transcript(mcp_client_session: ClientSession) -> None:
7069
async def test_get_transcript_with_language(mcp_client_session: ClientSession) -> None:
7170
video_id = "WjAXZkQSE2U"
7271

73-
title = fetch_title(video_id, "ja")
7472
expect = Transcript(
75-
title=title,
73+
title=fetch_title(video_id, "ja"),
7674
transcript="\n".join((item.text for item in YouTubeTranscriptApi().fetch(video_id, ["ja"]))),
7775
)
7876

@@ -96,9 +94,8 @@ async def test_get_transcript_fallback_language(
9694
) -> None:
9795
video_id = "LPZh9BOjkQs"
9896

99-
title = fetch_title(video_id, "en")
10097
expect = Transcript(
101-
title=title,
98+
title=fetch_title(video_id, "en"),
10299
transcript="\n".join((item.text for item in YouTubeTranscriptApi().fetch(video_id))),
103100
)
104101

@@ -140,9 +137,8 @@ async def test_get_transcript_not_found(mcp_client_session: ClientSession) -> No
140137
async def test_get_transcript_with_short_url(mcp_client_session: ClientSession) -> None:
141138
video_id = "LPZh9BOjkQs"
142139

143-
title = fetch_title(video_id, "en")
144140
expect = Transcript(
145-
title=title,
141+
title=fetch_title(video_id, "en"),
146142
transcript="\n".join((item.text for item in YouTubeTranscriptApi().fetch(video_id))),
147143
)
148144

0 commit comments

Comments
 (0)