tests/test_cli.py

# -*- coding: utf-8 -*-
from unittest import mock

from pytube import cli


@mock.patch("pytube.cli.YouTube")
@mock.patch("pytube.cli.sys")
def test_download(MockYouTube, mock_sys):
    instance = MockYouTube.return_value
    instance.prefetch_descramble.return_value = None
    instance.streams = mock.Mock()
    cli.download("asdf", 123)