code: def test_set_kwargs():
      with dask.config.set(foo__bar=1, foo__baz=2):
          assert config["foo"] == {"bar": 1, "baz": 2}
      assert "foo" not in config
  
      # Mix kwargs and dict, kwargs override
      with dask.config.set(
          {"foo.bar": 1, "foo.baz": 2}, foo__buzz=3, foo__bar=4
      ):
          assert config["foo"] == {"bar": 4, "baz": 2, "buzz": 3}
      assert "foo" not in config
  
      # Mix kwargs and nested dict, kwargs override
      with dask.config.set(
          {"foo": {"bar": 1, "baz": 2}}, foo__buzz=3, foo__bar=4
      ):
          assert config["foo"] == {"bar": 4, "baz": 2, "buzz": 3}
      assert "foo" not in config
  
imports: from __future__ import annotations
  import os
  import site
  import stat
  import sys
  from collections import OrderedDict
  from contextlib import contextmanager
  import pytest
  import yaml
  import dask.config
  from dask.config import _get_paths, canonical_name, collect, collect_env, collect_yaml, config, deserialize, ensure_file, expand_environment_variables, get, merge, pop, refresh, rename, serialize, update, update_defaults
  from dask.utils import tmpfile
label_desc: Use image node_id 21 for calling a global function with example usage: test_set_kwargs() without return types