@@ -234,7 +234,7 @@ def test_variable_types():
234
234
def test_make_entityset_default_args (teams_input ):
235
235
normalized_entityset = autonormalize .make_entityset (teams_input .get_df (), teams_input .get_deps ())
236
236
237
- dic_one = {'team_jersey_num ' : [0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ],
237
+ dic_one = {'jersey_num_team ' : [0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ],
238
238
'team' : ['Red' , 'Red' , 'Red' , 'Orange' , 'Orange' , 'Yellow' ,
239
239
'Yellow' , 'Green' , 'Green' , 'Blue' ],
240
240
'jersey_num' : [1 , 2 , 3 , 1 , 2 , 1 , 5 , 8 , 2 , 2 ],
@@ -254,7 +254,7 @@ def test_make_entityset_default_args(teams_input):
254
254
assert normalized_entityset .entities [2 ].df .equals (pd .DataFrame (
255
255
dic_three , index = ['austin' , 'boston' , 'chicago' , 'honolulu' ]))
256
256
257
- assert normalized_entityset .entities [0 ].variable_types ['team_jersey_num ' ] == Index
257
+ assert normalized_entityset .entities [0 ].variable_types ['jersey_num_team ' ] == Index
258
258
assert normalized_entityset .entities [0 ].variable_types ['team' ] == Id
259
259
assert normalized_entityset .entities [0 ].variable_types ['jersey_num' ] == Numeric
260
260
assert normalized_entityset .entities [0 ].variable_types ['player_name' ] == Categorical
@@ -272,7 +272,7 @@ def test_make_entityset_custom_args(teams_input):
272
272
name = 'Teams' ,
273
273
variable_types = {'state' : Text })
274
274
275
- dic_one = {'team_jersey_num ' : [0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ],
275
+ dic_one = {'jersey_num_team ' : [0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 , 9 ],
276
276
'team' : ['Red' , 'Red' , 'Red' , 'Orange' , 'Orange' , 'Yellow' ,
277
277
'Yellow' , 'Green' , 'Green' , 'Blue' ],
278
278
'jersey_num' : [1 , 2 , 3 , 1 , 2 , 1 , 5 , 8 , 2 , 2 ],
@@ -293,7 +293,7 @@ def test_make_entityset_custom_args(teams_input):
293
293
assert normalized_entityset .entities [2 ].df .equals (pd .DataFrame (
294
294
dic_three , index = ['austin' , 'boston' , 'chicago' , 'honolulu' ]))
295
295
296
- assert normalized_entityset .entities [0 ].variable_types ['team_jersey_num ' ] == Index
296
+ assert normalized_entityset .entities [0 ].variable_types ['jersey_num_team ' ] == Index
297
297
assert normalized_entityset .entities [0 ].variable_types ['team' ] == Id
298
298
assert normalized_entityset .entities [0 ].variable_types ['jersey_num' ] == Numeric
299
299
assert normalized_entityset .entities [0 ].variable_types ['player_name' ] == Categorical
0 commit comments