f2f419b89221602dc31c69de32df1cba281db481,upsetplot/tests/test_data.py,,test_from_contents,#,90

Before Change

    empty_data = pd.DataFrame(index=["aa", "bb", "cc", "dd", "ee", "ff"])
    out = from_contents(OrderedDict(contents), data=empty_data,
                        id_column=id_column)
    out2 = from_memberships(memberships=[{"cat1"},
                                         {"cat1"},
                                         {"cat1", "cat2"},
                                         {"cat2"},
                                         {"cat3"},
                                         []],
                            data=empty_data)
    assert_series_equal(out[id_column].reset_index(drop=True),
                        pd.Series(["aa", "bb", "cc", "dd", "ee", "ff"],
                                  name=id_column))
    assert_frame_equal(out.drop(columns=[id_column]), out2)

After Change

def test_from_contents(typ=set, id_column="id"):
    contents = {"cat1": {"aa", "bb", "cc"},
                "cat2": {"cc", "dd"},
                "cat3": {"ee"},
                }
    empty_data = pd.DataFrame(index=["aa", "bb", "cc", "dd", "ee"])
    baseline = from_contents(OrderedDict(contents), data=empty_data,
                             id_column=id_column)
    // data=None

Instances