65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
import json
|
|
from pytchat.parser.live import Parser
|
|
from pytchat.processors.speed.calculator import SpeedCalculator
|
|
|
|
parser = Parser(is_replay=False)
|
|
|
|
|
|
def test_speed_1(mocker):
|
|
'''test speed calculation with normal json.
|
|
test json has 15 chatdata, duration is 30 seconds,
|
|
so the speed of chatdata is 30 chats/minute.
|
|
'''
|
|
|
|
processor = SpeedCalculator(capacity=30)
|
|
|
|
_json = _open_file("tests/testdata/speed/speedtest_normal.json")
|
|
|
|
_, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
|
|
data = {
|
|
"video_id": "",
|
|
"timeout": 10,
|
|
"chatdata": chatdata
|
|
}
|
|
ret = processor.process([data])
|
|
assert 30 == ret
|
|
|
|
|
|
def test_speed_2(mocker):
|
|
'''test speed calculation with no valid chat data.
|
|
'''
|
|
processor = SpeedCalculator(capacity=30)
|
|
|
|
_json = _open_file("tests/testdata/speed/speedtest_undefined.json")
|
|
|
|
_, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
|
|
data = {
|
|
"video_id": "",
|
|
"timeout": 10,
|
|
"chatdata": chatdata
|
|
}
|
|
ret = processor.process([data])
|
|
assert ret == 0
|
|
|
|
|
|
def test_speed_3(mocker):
|
|
'''test speed calculation with empty data.
|
|
'''
|
|
processor = SpeedCalculator(capacity=30)
|
|
|
|
_json = _open_file("tests/testdata/speed/speedtest_empty.json")
|
|
|
|
_, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
|
|
data = {
|
|
"video_id": "",
|
|
"timeout": 10,
|
|
"chatdata": chatdata
|
|
}
|
|
ret = processor.process([data])
|
|
assert ret == 0
|
|
|
|
|
|
def _open_file(path):
|
|
with open(path, mode='r', encoding='utf-8') as f:
|
|
return f.read()
|