File size: 4,519 Bytes
ccd4117 328e22b d2c119a 328e22b ccd4117 8dfb028 ccd4117 328e22b d2c119a ccd4117 8b82476 328e22b d2c119a da19fc7 d2c119a 328e22b 8b82476 ccd4117 328e22b d2c119a 17564d0 ccd4117 328e22b d2c119a 17564d0 ccd4117 17564d0 328e22b d2c119a ccd4117 328e22b ccd4117 17564d0 328e22b d2c119a de6c1b0 ccd4117 328e22b ccd4117 de6c1b0 0649fcc de6c1b0 860893c 0649fcc ccd4117 17564d0 328e22b d2c119a de6c1b0 ccd4117 328e22b ccd4117 de6c1b0 0649fcc de6c1b0 860893c de6c1b0 ccd4117 328e22b d2c119a 17564d0 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 141 142 143 144 145 146 147 148 |
# -*- 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
@pytest.mark.skip(reason="Error with int sorting")
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(progressive=True)
.order_by('itag')
.all()
]
assert itags == ['17', '18', '22', '36', '43']
@pytest.mark.skip(reason="Error with int sorting")
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(progressive=True)
.order_by('itag')
.desc()
.all()
]
assert itags == ['43', '36', '22', '18', '17']
# non numerical values
mime_types = [
s.mime_type for s in cipher_signature.streams
.filter(progressive=True)
.order_by('mime_type')
.desc()
.all()
]
assert mime_types == [
'video/webm', 'video/mp4',
'video/mp4', 'video/3gpp', 'video/3gpp',
]
@pytest.mark.skip(reason="Error with int sorting")
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(progressive=True)
.order_by('itag')
.asc()
.all()
]
assert itags == ['17', '18', '22', '36', '43']
# non numerical values
mime_types = [
s.mime_type for s in cipher_signature.streams
.filter(progressive=True)
.order_by('mime_type')
.asc()
.all()
]
assert mime_types == [
'video/3gpp', 'video/3gpp',
'video/mp4', 'video/mp4', 'video/webm',
]
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)
|