def test_project_lambda_smoke_negatives():
eqo = EqualizedOdds()
events = ['label=False', 'label=True']
signs = ['+', '-']
labels = ['a', 'b']
midx = pd.MultiIndex.from_product([signs, events, labels],
names=[_SIGN, _EVENT, _GROUP_ID])
df = pd.DataFrame()
# Note that the '-' labels are larger
df = 0 + pd.Series([1, 2, 11, 19, 1001, 1110, 1230, 1350], index=midx)
ls = eqo.project_lambda(df)
expected = pd.DataFrame()
expected = 0 + pd.Series([0, 0, 0, 0, 1000, 1108, 1219, 1331], index=midx)
assert expected.equals(ls)
def test_project_lambda_smoke_positives():
# This is a repeat of the _negatives method but with
# the '+' indices larger
eqo = EqualizedOdds()
events = ['label=False', 'label=True']
signs = ['+', '-']
labels = ['a', 'b']
midx = pd.MultiIndex.from_product([signs, events, labels],
names=[_SIGN, _EVENT, _GROUP_ID])
df = pd.DataFrame()
# Note that the '-' indices are now smaller
df = 0 + pd.Series([200, 300, 100, 600, 4, 5, 6, 7], index=midx)
ls = eqo.project_lambda(df)
expected = pd.DataFrame()
expected = 0 + pd.Series([196, 295, 94, 593, 0, 0, 0, 0], index=midx)
assert expected.equals(ls)