def test_direct_copy(games_es): home_team = next(r for r in games_es.relationships if r.child_variable.id == 'home_team_id') feat = DirectFeature(games_es['teams']['name'], games_es['games'], relationship=home_team) copied = feat.copy() assert copied.entity == feat.entity assert copied.base_features == feat.base_features assert copied.relationship_path == feat.relationship_path
def test_direct_copy(games_es): home_team = next(r for r in games_es.relationships if r._child_column_name == 'home_team_id') feat = DirectFeature(IdentityFeature(games_es['teams'].ww['name']), 'games', relationship=home_team) copied = feat.copy() assert copied.dataframe_name == feat.dataframe_name assert copied.base_features == feat.base_features assert copied.relationship_path == feat.relationship_path
def test_direct_copy(games_es): home_team = next(r for r in games_es.relationships if r._child_column_name == "home_team_id") feat = DirectFeature(IdentityFeature(games_es["teams"].ww["name"]), "games", relationship=home_team) copied = feat.copy() assert copied.dataframe_name == feat.dataframe_name assert copied.base_features == feat.base_features assert copied.relationship_path == feat.relationship_path