Saverr/svtplay-dl
1
0
Fork 0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-27 21:54:17 +01:00
Code Issues Projects Releases Wiki Activity

protocol_prio: Make the test suite import module from right location

Browse Source 
The tests are still failing, but not from ImportError anymore.
This commit is contained in:
Olof Johansson 2018-09-08 22:21:54 +02:00 committed by Johan Andersson
parent 559e950e68
commit 5f087f0efa
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/svtplay_dl/tests/protocol_prio.py
View File

@ -4,16 +4,14 @@
from __future__ import absolute_import
import unittest
from svtplay_dl.utils import protocol_prio
from svtplay_dl.utils.stream import protocol_prio
class Stream(object):
class VideoRetriever(object):
def __init__(self, proto, bitrate):
self.proto = proto
self.bitrate = bitrate
def name(self):
return self.proto
self.name = proto
def __repr__(self):
return '%s(%d)' % (self.proto.upper(), self.bitrate)
@ -21,11 +19,11 @@ class Stream(object):
class PrioStreamsTest(unittest.TestCase):
def _gen_proto_case(self, ordered, unordered, expected=None):
streams = [Stream(x, 100) for x in unordered]
streams = [VideoRetriever(x, 100) for x in unordered]
kwargs = {}
if expected is None:
expected = [str(Stream(x, 100)) for x in ordered]
expected = [str(VideoRetriever(x, 100)) for x in ordered]
return self.assertEqual(
[str(x) for x in protocol_prio(streams, ordered, **kwargs)],