Ejemplos de Check.remove_columns en Python

Ejemplo n.º 1

Archivo: test_modules.py Proyecto: vandana-11/cognito

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'
    ]

Ejemplo n.º 2

Archivo: test_modules.py Proyecto: AishikaBanik98/cognito

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

Ejemplo n.º 3

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']