adb2f55d8629b588bba17a5232946b2245c157fc,tests/test_preprocessing.py,,encoded_data,#Any#,32

Before Change

def encoded_data(data):
    dat_cat = data.select_dtypes(include=["category"])

    expected_cat = {}
    for nam, col in dat_cat.iteritems():
        for cat in col.cat.categories[1:]:
            expected_cat["{}={}".format(nam, cat)] = pd.Series(
                col == cat, dtype=np.float64)
    expected_cat = pd.DataFrame(expected_cat)

    expected_data = pd.concat((data.drop(dat_cat, axis=1), expected_cat), axis=1)
    return expected_data


class TestOneHotEncoder(TestCase):

After Change

def encoded_data(data):
    expected = []
    for nam, col in data.iteritems():
        if hasattr(col, "cat"):
            for cat in col.cat.categories[1:]:
                name = "{}={}".format(nam, cat)
                s = pd.Series(col == cat, dtype=np.float64)
                expected.append((name, s))
        else:
            expected.append((nam, col))

    expected_data = pd.DataFrame.from_items(expected)
    return expected_data

Instances