L ivddlmZddlmZddlmZmZddlmZerddl m Z m Z e d dZ d dZ y) ) annotations)contextmanager) TYPE_CHECKINGAny)config)Callable Generatorc#Kddlm}t|}|jtd|5ddddy#1swYyxYww)aA context manager that overrides config options. It can also be used as a function decorator. Examples -------- >>> with patch_config_options({"server.headless": True}): ... assert config.get_option("server.headless") is True ... # Other test code that relies on these options >>> @patch_config_options({"server.headless": True}) ... def test_my_thing(): ... assert config.get_option("server.headless") is True r)patch get_option)newN) unittest.mockr build_mock_config_get_optionobjectr)config_overridesr mock_get_options _/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/streamlit/testing/v1/util.pypatch_config_optionsrs?$$23CDO fl @ s*A: AAAc8tjdfd }|S)Nc&|vr|S|S)N)nameorig_get_optionoverrides_dicts rmock_config_get_optionz.mock_config_get_option8s! > !!$' 't$$)rstrreturnr)rr )rrrs` @rrr3s''O% "!rN)rdict[str, Any]rzGenerator[None, None, None])rrrzCallable[[str], Any]) __future__r contextlibrtypingrr streamlitrcollections.abcrr rrrrrr%sR#%%3$ 0 "" " "r