Example #1
0
def test_get_X_columns_deletion():
    df = get_raw_from_github(url=url)
    X = get_X(df)
    expected_df = generate_raw_df_test()
    del expected_df["DEFAULT"]
    del expected_df["ID"]
    assert_frame_equal(X, expected_df)
Example #2
0
def test_get_X_qty_columns():
    df = get_raw_from_github(url=url)
    X = get_X(df)
    expected_columns = 23
    assert len(X.columns) == expected_columns
Example #3
0
def test_get_X_check_is_dataframe():
    df = get_raw_from_github(url=url)
    X = get_X(df)
    assert isinstance(X, pd.DataFrame) == True
Example #4
0
def test_get_y_check_default_column_is_dataframe():
    df = get_raw_from_github(url=url)
    y = get_y(df)
    assert isinstance(y, pd.DataFrame) == True
Example #5
0
def test_get_y_check_default_column():
    df = get_raw_from_github(url=url)
    y = get_y(df)
    assert y.columns[0] == "DEFAULT"
Example #6
0
def test_get_raw_from_github():
    df = get_raw_from_github(url=url)
    expected_df = generate_raw_df_test()
    assert_frame_equal(df, expected_df)
Example #7
0
def test_get_raw_from_github_if_is_a_dataframe():
    df = get_raw_from_github(url=url)
    assert isinstance(df, pd.DataFrame) == True