def test_resolve_stream_name(self):
a = Mock()
b = Mock()
c = Mock()
d = Mock()
e = Mock()
streams = {
"160p": a,
"360p": b,
"480p": c,
"720p": d,
"1080p": e,
"worst": b,
"best": d,
"worst-unfiltered": a,
"best-unfiltered": e
}
self.assertEqual(resolve_stream_name(streams, "unknown"), "unknown")
self.assertEqual(resolve_stream_name(streams, "160p"), "160p")
self.assertEqual(resolve_stream_name(streams, "360p"), "360p")
self.assertEqual(resolve_stream_name(streams, "480p"), "480p")
self.assertEqual(resolve_stream_name(streams, "720p"), "720p")
self.assertEqual(resolve_stream_name(streams, "1080p"), "1080p")
self.assertEqual(resolve_stream_name(streams, "worst"), "360p")
self.assertEqual(resolve_stream_name(streams, "best"), "720p")
self.assertEqual(resolve_stream_name(streams, "worst-unfiltered"), "160p")
self.assertEqual(resolve_stream_name(streams, "best-unfiltered"), "1080p")
def test_resolve_stream_name(self):
high = Mock()
medium = Mock()
low = Mock()
streams = {
"low": low,
"medium": medium,
"high": high,
"worst": low,
"best": high
}
self.assertEqual("high", resolve_stream_name(streams, "best"))
self.assertEqual("low", resolve_stream_name(streams, "worst"))
self.assertEqual("medium", resolve_stream_name(streams, "medium"))
self.assertEqual("high", resolve_stream_name(streams, "high"))
self.assertEqual("low", resolve_stream_name(streams, "low"))
