def test_get_numeric_subset(): df = iris.copy() df['e'] = df['e'].astype(str) subset = get_numeric_columns(df) assert subset.shape != df.shape
def test_get_numeric(): subset = get_numeric_columns(iris) assert subset.equals(iris) assert subset is not iris