69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
import json
|
|
from pytchat.parser.live import Parser
|
|
from pytchat.processors.superchat.calculator import SuperchatCalculator
|
|
from pytchat.exceptions import ChatParseException
|
|
parse = SuperchatCalculator()._parse
|
|
|
|
|
|
def _open_file(path):
|
|
with open(path,mode ='r',encoding = 'utf-8') as f:
|
|
return f.read()
|
|
|
|
def load_chatdata(filepath):
|
|
parser = Parser(is_replay=True)
|
|
#print(json.loads(_open_file(filepath)))
|
|
contents = parser.get_contents( json.loads(_open_file(filepath)))
|
|
return parser.parse(contents)[1]
|
|
|
|
|
|
|
|
def test_parse_1():
|
|
renderer ={"purchaseAmountText":{"simpleText":"￥2,000"}}
|
|
symbol ,amount = parse(renderer)
|
|
assert symbol == '￥'
|
|
assert amount == 2000.0
|
|
|
|
def test_parse_2():
|
|
renderer ={"purchaseAmountText":{"simpleText":"ABC\x0a200"}}
|
|
symbol ,amount = parse(renderer)
|
|
assert symbol == 'ABC\x0a'
|
|
assert amount == 200.0
|
|
|
|
def test_process_0():
|
|
"""
|
|
parse superchat data
|
|
"""
|
|
chat_component = {
|
|
'video_id':'',
|
|
'timeout':10,
|
|
'chatdata':load_chatdata(r"tests\testdata\calculator\superchat_0.json")
|
|
}
|
|
assert SuperchatCalculator().process([chat_component])=={'￥': 6800.0, '€': 2.0}
|
|
|
|
def test_process_1():
|
|
"""
|
|
parse no superchat data
|
|
"""
|
|
chat_component = {
|
|
'video_id':'',
|
|
'timeout':10,
|
|
'chatdata':load_chatdata(r"tests\testdata\calculator\text_only.json")
|
|
}
|
|
assert SuperchatCalculator().process([chat_component])=={}
|
|
|
|
def test_process_2():
|
|
"""
|
|
try to parse after replay end
|
|
"""
|
|
try:
|
|
chat_component = {
|
|
'video_id':'',
|
|
'timeout':10,
|
|
'chatdata':load_chatdata(r"tests\testdata\calculator\replay_end.json")
|
|
}
|
|
assert False
|
|
SuperchatCalculator().process([chat_component])
|
|
except ChatParseException:
|
|
assert True
|
|
|