@@ -171,17 +171,12 @@ def test_info(model_filename):
171
171
with Tagger ().open (model_filename ) as tagger :
172
172
res = tagger .info ()
173
173
174
- assert res .transitions [("sunny" , "sunny" )] > res .transitions [("sunny" , "rainy" )]
175
- assert (
176
- res .state_features [("walk" , "sunny" )]
177
- > res .state_features [("walk" , "rainy" )]
178
- )
179
- assert ("солнце:не светит" , "rainy" ) in res .state_features
180
- assert res .header ["num_labels" ] == "2"
181
- assert set (res .labels .keys ()) == {"sunny" , "rainy" }
182
- assert set (res .attributes .keys ()) == {
183
- "shop" , "walk" , "clean" , "солнце:не светит"
184
- }
174
+ assert res .transitions [('sunny' , 'sunny' )] > res .transitions [('sunny' , 'rainy' )]
175
+ assert res .state_features [('walk' , 'sunny' )] > res .state_features [('walk' , 'rainy' )]
176
+ assert (u'солнце:не светит' , u'rainy' ) in res .state_features
177
+ assert res .header ['num_labels' ] == '3'
178
+ assert set (res .labels .keys ()) == set (['sunny' , 'rainy' , '好' ])
179
+ assert set (res .attributes .keys ()) == set (['shop' , 'walk' , 'clean' , u'солнце:не светит' , 'world' ])
185
180
186
181
# it shouldn't segfault on a closed tagger
187
182
with pytest .raises (RuntimeError ):
0 commit comments