@@ -123,13 +123,35 @@ def test_parse_events_signature(self):
123
123
with self .assertRaises (TypeError ):
124
124
parser .parse_events (5 )
125
125
126
- # def parse_voice(self) -> Dict[str, bytes]: ...
127
- # def parse_ticks(
128
- # self,
129
- # wanted_props: Sequence[str],
130
- # player: Optional[Sequence[int]] = None,
131
- # ticks: Optional[Sequence[int]] = None,
132
- # ) -> pd.DataFrame:
126
+ def test_parse_voice_signature (self ):
127
+ parser = DemoParser ("tests/data/test.dem" )
128
+ voice = parser .parse_voice ()
129
+ self .assertIsInstance (voice , dict )
130
+ for key , value in voice .items ():
131
+ self .assertIsInstance (key , str )
132
+ self .assertIsInstance (value , bytes )
133
+
134
+ def test_parse_ticks_signature (self ):
135
+ parser = DemoParser ("tests/data/test.dem" )
136
+
137
+ ticks = parser .parse_ticks (["X" , "Y" ])
138
+ self .assertIsInstance (ticks , pd .DataFrame )
139
+
140
+ parser .parse_ticks (["X" , "Y" ], players = [1 , 2 , 3 ], ticks = [1 , 2 , 3 ])
141
+ parser .parse_ticks (["X" , "Y" ], players = None , ticks = None )
142
+ parser .parse_ticks (["X" , "Y" ], players = [], ticks = [])
143
+
144
+ with self .assertRaises (TypeError ):
145
+ parser .parse_ticks (["X" , "Y" ], players = 5 , ticks = None )
146
+
147
+ with self .assertRaises (TypeError ):
148
+ parser .parse_ticks (["X" , "Y" ], players = None , ticks = 5 )
149
+
150
+ with self .assertRaises (TypeError ):
151
+ parser .parse_ticks (["X" , "Y" ], players = "Test" )
152
+
153
+ with self .assertRaises (TypeError ):
154
+ parser .parse_ticks (5 )
133
155
134
156
135
157
if __name__ == "__main__" :
0 commit comments