def test_isnull_feat(es): value = IdentityFeature(es['log']['value']) diff = Diff(value, es['log']['session_id']) isnull = IsNull(diff) features = [isnull] pandas_backend = PandasBackend(es, features) df = pandas_backend.calculate_all_features(range(15), None) # correct_vals_diff = [ # np.nan, 5, 5, 5, 5, np.nan, 1, 1, 1, np.nan, np.nan, 5, np.nan, 7, 7] correct_vals = [True, False, False, False, False, True, False, False, False, True, True, False, True, False, False] values = df[isnull.get_name()].values.tolist() assert correct_vals == values
def test_call_trans(): primitive = IsNull() for _ in range(2): assert pd.Series([False] * 6).equals(primitive(range(6)))
def is_null(self): """Compares feature to null by equality""" from featuretools.primitives import IsNull return IsNull(self)
def test_call_trans(): primitive = IsNull() assert pd.Series([False for i in range(6)]).equals(primitive(range(6))) assert pd.Series([False for i in range(6)]).equals(primitive(range(6)))