def test_remove_columns_1(): df = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman.csv')) check = Check() assert list(check.remove_columns(df).columns) == [ 'Location', 'population', 'nonwhite', 'density', 'crime' ]
def test_remove_columns(): df = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman.csv')) df1 = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman_remove_columns.csv')) check = Check() assert check.remove_columns(df).equals(df1) == True
def test_remove_columns_2(): df = pd.read_csv(os.path.join(os.path.dirname(__file__), 'data', 'msleep_ggplot.csv')) check = Check() assert list(check.remove_columns(df).columns) == ['name', 'genus', 'vore', 'order', 'conservation', 'sleep_total', 'sleep_rem', 'awake', 'brainwt', 'bodywt']