Implement raise_for_status()

tests/test_speed_calculator.py

@@ -1,15 +1,9 @@
 import json
-import pytest
-import asyncio,aiohttp
 from pytchat.parser.live import Parser
-from pytchat.processors.compatible.processor import CompatibleProcessor
-from pytchat.exceptions import (
-    NoLivechatRendererException,NoYtinitialdataException,
-    ResponseContextError, NoContentsException)
-
 from pytchat.processors.speed.calculator import SpeedCalculator
 
-parser = Parser(is_replay =False)
+parser = Parser(is_replay=False)
+
 
 def test_speed_1(mocker):
     '''test speed calculation with normal json.
@@ -23,13 +17,14 @@ def test_speed_1(mocker):
 
     _, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
     data = {
-        "video_id" : "",
-        "timeout" : 10,
-        "chatdata" : chatdata
+        "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.
     '''
@@ -39,13 +34,14 @@ def test_speed_2(mocker):
 
     _, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
     data = {
-        "video_id" : "",
-        "timeout" : 10,
-        "chatdata" : chatdata
+        "video_id": "",
+        "timeout": 10,
+        "chatdata": chatdata
     }
     ret = processor.process([data])
-    assert 0 == ret
-    
+    assert ret == 0
+
+
 def test_speed_3(mocker):
     '''test speed calculation with empty data.
     '''
@@ -55,14 +51,14 @@ def test_speed_3(mocker):
 
     _, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
     data = {
-        "video_id" : "",
-        "timeout" : 10,
-        "chatdata" : chatdata
+        "video_id": "",
+        "timeout": 10,
+        "chatdata": chatdata
     }
     ret = processor.process([data])
-    assert 0 == ret
-    
+    assert ret == 0
+
 
 def _open_file(path):
-    with open(path,mode ='r',encoding = 'utf-8') as f:
+    with open(path, mode='r', encoding='utf-8') as f:
         return f.read()