code: def test_remove_feature():
      remove_fn = transform_lib.remove_feature("a")
      assert list(remove_fn(example={"a": 1, "b": 2})) == [{"b": 2}]
  
file_path: datasets/tensorflow_datasets/core/transform/transform_lib_test.py
imports: import functools
  from typing import Iterator
  import pytest
  from tensorflow_datasets.core.transform import transform_lib
label_desc: Use image node_id 12 for calling a global function with example usage: test_remove_feature() without return types