File size: 4,436 Bytes
ccd4117 328e22b d2c119a 328e22b ccd4117 3047241 ccd4117 328e22b d2c119a 3047241 ccd4117 3047241 328e22b d2c119a da19fc7 d2c119a 3047241 8b82476 ccd4117 328e22b d2c119a 17564d0 ccd4117 328e22b d2c119a 17564d0 ccd4117 328e22b d2c119a ccd4117 3047241 ccd4117 3047241 ccd4117 328e22b d2c119a de6c1b0 ccd4117 3047241 ccd4117 3047241 ccd4117 3047241 de6c1b0 3047241 0649fcc de6c1b0 3047241 0649fcc ccd4117 328e22b d2c119a de6c1b0 ccd4117 3047241 ccd4117 3047241 ccd4117 3047241 de6c1b0 3047241 0649fcc de6c1b0 3047241 de6c1b0 ccd4117 328e22b d2c119a 17564d0 3047241 ccd4117 328e22b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 |
# -*- coding: utf-8 -*-
"""Unit tests for the :class:`StreamQuery <StreamQuery>` class."""
import pytest
def test_count(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.count` returns an accurate amount."""
assert cipher_signature.streams.count() == 22
@pytest.mark.parametrize(
"test_input,expected",
[
({"progressive": True}, [18]),
({"resolution": "720p"}, [136, 247]),
({"res": "720p"}, [136, 247]),
({"fps": 30, "resolution": "480p"}, [135, 244]),
({"mime_type": "audio/mp4"}, [140]),
({"type": "audio"}, [140, 249, 250, 251]),
({"subtype": "3gpp"}, []),
({"abr": "128kbps"}, [140]),
({"bitrate": "128kbps"}, [140]),
({"audio_codec": "opus"}, [249, 250, 251]),
({"video_codec": "vp9"}, [248, 247, 244, 243, 242, 278]),
({"only_audio": True}, [140, 249, 250, 251]),
({"only_video": True, "video_codec": "avc1.4d4015"}, [133]),
({"adaptive": True, "resolution": "1080p"}, [137, 248]),
({"custom_filter_functions": [lambda s: s.itag == 18]}, [18]),
],
)
def test_filters(test_input, expected, cipher_signature):
"""Ensure filters produce the expected results."""
result = [s.itag for s in cipher_signature.streams.filter(**test_input).all()]
assert result == expected
@pytest.mark.parametrize("test_input", ["first", "last"])
def test_empty(test_input, cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.last` and
:meth:`~pytube.StreamQuery.first` return None if the resultset is
empty.
"""
query = cipher_signature.streams.filter(video_codec="vp20")
fn = getattr(query, test_input)
assert fn() is None
def test_get_last(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.last` returns the expected
:class:`Stream <Stream>`.
"""
assert cipher_signature.streams.last().itag == 251
def test_get_first(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.first` returns the expected
:class:`Stream <Stream>`.
"""
assert cipher_signature.streams.first().itag == 18
def test_order_by(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.order_by` sorts the list of
:class:`Stream <Stream>` instances in the expected order.
"""
itags = [
s.itag
for s in cipher_signature.streams.filter(type="audio").order_by("itag").all()
]
assert itags == [140, 249, 250, 251]
def test_order_by_descending(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.desc` sorts the list of
:class:`Stream <Stream>` instances in the reverse order.
"""
# numerical values
itags = [
s.itag
for s in cipher_signature.streams.filter(type="audio")
.order_by("itag")
.desc()
.all()
]
assert itags == [251, 250, 249, 140]
def test_order_by_non_numerical(cipher_signature):
mime_types = [
s.mime_type
for s in cipher_signature.streams.filter(res="360p")
.order_by("mime_type")
.desc()
.all()
]
assert mime_types == ["video/webm", "video/mp4", "video/mp4"]
def test_order_by_ascending(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.desc` sorts the list of
:class:`Stream <Stream>` instances in ascending order.
"""
# numerical values
itags = [
s.itag
for s in cipher_signature.streams.filter(type="audio")
.order_by("itag")
.asc()
.all()
]
assert itags == [140, 249, 250, 251]
def test_order_by_non_numerical_ascending(cipher_signature):
mime_types = [
s.mime_type
for s in cipher_signature.streams.filter(res="360p")
.order_by("mime_type")
.asc()
.all()
]
assert mime_types == ["video/mp4", "video/mp4", "video/webm"]
def test_order_by_with_none_values(cipher_signature):
abrs = [s.abr for s in cipher_signature.streams.order_by("abr").asc().all()]
assert abrs == ["50kbps", "70kbps", "96kbps", "128kbps", "160kbps"]
def test_get_by_itag(cipher_signature):
"""Ensure :meth:`~pytube.StreamQuery.get_by_itag` returns the expected
:class:`Stream <Stream>`.
"""
assert cipher_signature.streams.get_by_itag(18).itag == 18
assert cipher_signature.streams.get_by_itag("18").itag == 18
def test_get_by_non_existent_itag(cipher_signature):
assert not cipher_signature.streams.get_by_itag(22983)
|