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
Esempio n. 3
0
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