def test_refresh():
defaults = []
config = {}
update_defaults({"a": 1}, config=config, defaults=defaults)
assert config == {"a": 1}
refresh(paths=[], env={"DASK_B": "2"}, config=config, defaults=defaults)
assert config == {"a": 1, "b": 2}
refresh(paths=[], env={"DASK_C": "3"}, config=config, defaults=defaults)
assert config == {"a": 1, "c": 3}
def test_refresh():
defaults = []
config = {}
update_defaults({'a': 1}, config=config, defaults=defaults)
assert config == {'a': 1}
refresh(paths=[], env={'DASK_B': '2'}, config=config, defaults=defaults)
assert config == {'a': 1, 'b': 2}
refresh(paths=[], env={'DASK_C': '3'}, config=config, defaults=defaults)
assert config == {'a': 1, 'c': 3}
def test_refresh():
defaults = []
config = {}
update_defaults({'a': 1}, config=config, defaults=defaults)
assert config == {'a': 1}
refresh(paths=[], env={'DASK_B': '2'}, config=config, defaults=defaults)
assert config == {'a': 1, 'b': 2}
refresh(paths=[], env={'DASK_C': '3'}, config=config, defaults=defaults)
assert config == {'a': 1, 'c': 3}
def test_custom_yaml(tmpdir):
custom_config = {}
custom_config["sql"] = dask_config.get("sql")
custom_config["sql"]["groupby"]["split_out"] = 16
custom_config["sql"]["foo"] = {"bar": [1, 2, 3], "baz": None}
with open(os.path.join(tmpdir, "custom-sql.yaml"), mode="w") as f:
yaml.dump(custom_config, f)
dask_config.refresh(
paths=[tmpdir]) # Refresh config to read from updated environment
assert custom_config["sql"] == dask_config.get("sql")
dask_config.refresh()
def test_dask_setconfig():
dask_config.set({"sql.foo.bar": 1})
with dask_config.set({"sql.foo.baz": "2"}):
assert dask_config.get("sql.foo") == {"bar": 1, "baz": "2"}
assert dask_config.get("sql.foo") == {"bar": 1}
dask_config.refresh()
def test_env_variable():
with mock.patch.dict("os.environ",
{"DASK_SQL__GROUPBY__SPLIT_OUT": "200"}):
dask_config.refresh()
assert dask_config.get("sql.groupby.split-out") == 200
dask_config.refresh()