added tests

tests/test_cli.py

@@ -153,12 +153,27 @@ def test_parse_args_truthy():
             "en",
             "-l",
             "--itag=10",
+            "-vvv",
         ],
     )
     assert args.url == "http://youtube.com/watch?v=9bZkp7q19f0"
     assert args.build_playback_report is True
     assert args.itag == 10
     assert args.list is True
+    assert args.verbosity == 3
+
+
+@mock.patch("pytube.cli.setup_logger", return_value=None)
+def test_main_logging_setup(setup_logger):
+    # Given
+    parser = argparse.ArgumentParser()
+    args = parse_args(parser, ["http://fakeurl", "-v"])
+    cli._parse_args = MagicMock(return_value=args)
+    # When
+    with pytest.raises(SystemExit):
+        cli.main()
+    # Then
+    setup_logger.assert_called_with(40)
 
 
 @mock.patch("pytube.cli.YouTube", return_value=None)

tests/test_helpers.py

@@ -5,7 +5,7 @@ import pytest
 
 from pytube import helpers
 from pytube.exceptions import RegexMatchError
-from pytube.helpers import deprecated, cache, target_directory
+from pytube.helpers import deprecated, cache, target_directory, setup_logger
 
 
 def test_regex_search_no_match():
@@ -74,3 +74,15 @@ def test_target_directory_with_absolute_path(_, makedirs):
 def test_target_directory_with_no_path(_, makedirs):
     assert target_directory() == "/cwd"
     makedirs.assert_called()
+
+
+@mock.patch("pytube.helpers.logging")
+def test_setup_logger(logging):
+    # Given
+    logger = logging.getLogger.return_value
+    # When
+    setup_logger(20)
+    # Then
+    logging.getLogger.assert_called_with("pytube")
+    logger.addHandler.assert_called()
+    logger.setLevel.assert_called_with(20)