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"))