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