def test_format_valid_streams(self):
a = Mock()
b = Mock()
c = Mock()
streams = {"audio": a, "720p": b, "1080p": c, "worst": b, "best": c}
self.assertEqual(format_valid_streams(_TestPlugin, streams),
", ".join(["audio", "720p (worst)", "1080p (best)"]))
streams = {
"audio": a,
"720p": b,
"1080p": c,
"worst-unfiltered": b,
"best-unfiltered": c
}
self.assertEqual(
format_valid_streams(_TestPlugin, streams), ", ".join([
"audio", "720p (worst-unfiltered)", "1080p (best-unfiltered)"
]))
def test_format_valid_streams(self):
class FakePlugin:
@classmethod
def stream_weight(cls, stream):
return Plugin.stream_weight(stream)
a = Mock()
b = Mock()
c = Mock()
streams = {
"audio": a,
"720p": b,
"1080p": c,
"worst": b,
"best": c
}
self.assertEqual(
format_valid_streams(FakePlugin, streams),
", ".join([
"audio",
"720p (worst)",
"1080p (best)"
])
)
streams = {
"audio": a,
"720p": b,
"1080p": c,
"worst-unfiltered": b,
"best-unfiltered": c
}
self.assertEqual(
format_valid_streams(FakePlugin, streams),
", ".join([
"audio",
"720p (worst-unfiltered)",
"1080p (best-unfiltered)"
])
)
