def test_drop_missing_greater(): expected = ['country', 'size'] output = list(feature_engineering.drop_missing(df2, 90).columns) assert expected == output
def test_drop_missing(): expected = ['country'] output = list(feature_engineering.drop_missing(df2, 75).columns) assert expected == output