code: def test_remove_field_transform():
      one_removed = transform_schema(
          TestSchema,
          TransformSpec(
              lambda x: x, edit_fields=None, removed_fields=["int"]
          ),
      )
      assert set(one_removed.fields.keys()) == {"string", "double"}
  
      two_removed = transform_schema(
          TestSchema,
          TransformSpec(
              lambda x: x,
              edit_fields=None,
              removed_fields=["int", "double"],
          ),
      )
      assert set(two_removed.fields.keys()) == {"string"}
  
file_path: petastorm/petastorm/tests/test_transform.py
imports: import numpy
  import pytest
  from petastorm.transform import edit_field, transform_schema, TransformSpec
  from petastorm.unischema import Unischema, UnischemaField
label_desc: Use image node_id 2 for calling a global function with example usage: test_remove_field_transform() without return types