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