code: def test_is_small_dataset(
      dataset_size, input_in_memory_max_size, monkeypatch
  ):
      if input_in_memory_max_size != "default":
          monkeypatch.setattr(
              datasets.config,
              "IN_MEMORY_MAX_SIZE",
              input_in_memory_max_size,
          )
      in_memory_max_size = datasets.config.IN_MEMORY_MAX_SIZE
      if input_in_memory_max_size == "default":
          assert in_memory_max_size == 0
      else:
          assert in_memory_max_size == input_in_memory_max_size
      if dataset_size and in_memory_max_size:
          expected = dataset_size < in_memory_max_size
      else:
          expected = False
      result = is_small_dataset(dataset_size)
      assert result == expected
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_is_small_dataset(dataset_size, input_in_memory_max_size, monkeypatch) without return types