def test_project_away_badcolumn(): df = pd.DataFrame() df["A"] = [1, 2] w = Wrap(df) with pytest.raises(KeyError): w.project_away("B")
def test_project_away_wildcard_nomatch(): df = pd.DataFrame() df["A"] = [1, 2] w = Wrap(df) with pytest.raises(KeyError): w.project_away("A, B*")
def test_project_away_wildcard(): df = create_df() df = df[["A", "B"]] df["AA"] = df["B"] w = Wrap(df) wnew = w.project_away("A*") assert ["B"] == list(wnew.df.columns)
def test_project_away(self): df = create_df() df = df[["A", "B", "C"]] w = Wrap(df) wnew = w.project_away("B") self.assertListEqual(["A", "C"], list(wnew.df.columns)) self.assertListEqual(["A", "B", "C"], list(w.df.columns))
def test_project_away_wildcard2(): df = pd.DataFrame() df["A"] = [1, 2] df["ABC"] = [1, 2] df["B"] = [1, 2] df["Q"] = [1, 2] df["b"] = [1, 2] df["hiA"] = [1, 2] df["hiZ"] = [1, 2] df["hiZ2"] = [1, 2] w = Wrap(df) wnew = w.project_away("*B*, A*, *Z") assert ["Q", "b", "hiA", "hiZ2"] == list(wnew.df.columns)