def test_compare_all_nans(es):
nan_feat = Mode(es['log']['product_id'], es['sessions'])
compare = nan_feat == 'brown bag'
# before all data
time_last = pd.Timestamp('1/1/1993')
pandas_backend = PandasBackend(es, [nan_feat, compare])
df = pandas_backend.calculate_all_features(instance_ids=[0, 1, 2],
time_last=time_last)
assert df[nan_feat.get_name()].dropna().shape[0] == 0
assert not df[compare.get_name()].any()
def test_compare_all_nans(es):
nan_feat = Mode(es['log']['product_id'], es['sessions'])
compare = nan_feat == 'brown bag'
# before all data
time_last = pd.Timestamp('1/1/1993')
pandas_backend = PandasBackend(es, [nan_feat, compare])
df = pandas_backend.calculate_all_features(instance_ids=[0, 1, 2],
time_last=time_last)
assert df[nan_feat.get_name()].dropna().shape[0] == 0
assert not df[compare.get_name()].any()
def test_make_agg_feat_multiple_dtypes(entityset, backend):
compare_prod = IdentityFeature(entityset['log']['product_id']) == 'coke zero'
agg_feat = Count(entityset['log']['id'],
parent_entity=entityset['sessions'],
where=compare_prod)
agg_feat2 = Mode(entityset['log']['product_id'],
parent_entity=entityset['sessions'],
where=compare_prod)
pandas_backend = backend([agg_feat, agg_feat2])
df = pandas_backend.calculate_all_features(instance_ids=[0],
time_last=None)
v = df[agg_feat.get_name()][0]
v2 = df[agg_feat2.get_name()][0]
assert (v == 3)
assert (v2 == 'coke zero')
def test_make_agg_feat_multiple_dtypes(entityset, backend):
compare_prod = IdentityFeature(entityset['log']['product_id']) == 'coke zero'
agg_feat = Count(entityset['log']['id'],
parent_entity=entityset['sessions'],
where=compare_prod)
agg_feat2 = Mode(entityset['log']['product_id'],
parent_entity=entityset['sessions'],
where=compare_prod)
pandas_backend = backend([agg_feat, agg_feat2])
df = pandas_backend.calculate_all_features(instance_ids=[0],
time_last=None)
v = df[agg_feat.get_name()][0]
v2 = df[agg_feat2.get_name()][0]
assert (v == 3)
assert (v2 == 'coke zero')