Skip to content

Commit 6ed06e7

Browse files
committed
- Add test for video_transcription command;
- Add some necessary improvements in other files
1 parent 7f7e633 commit 6ed06e7

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

tests/commands/test_channel_info.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,4 +52,6 @@ def test_channel_ids_from_urls_and_usernames(mocker, channels_urls, usernames):
5252
channel_id_from_username_mock.assert_has_calls(
5353
[call(username) for username in usernames]
5454
)
55-
channels_infos_mock.assert_called_once_with([ids_from_urls_mock, ids_from_usernames_mock])
55+
channels_infos_mock.assert_called_once()
56+
assert ids_from_usernames_mock in channels_infos_mock.call_args.args[0]
57+
assert ids_from_urls_mock in channels_infos_mock.call_args.args[0]

youtool/commands/base.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,11 @@ def execute(cls, **kwargs) -> str: # noqa: D417
8080
raise NotImplementedError()
8181

8282
@staticmethod
83-
def data_from_csv(file_path: Path, data_column_name: Optional[str] = None) -> List[str]:
83+
def data_from_csv(
84+
file_path: Path,
85+
data_column_name: Optional[str] = None,
86+
raise_column_exception: bool = True
87+
) -> List[str]:
8488
"""Extracts a list of URLs from a specified CSV file.
8589
8690
Args:

0 commit comments

Comments
 (0)