scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

parameterize test

Browse source
This commit is contained in:
Ryan Crabbe 2026-02-25 17:30:59 -08:00
parent 9157c1ebe2
commit bfceefbb26
1 changed files with 10 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
tests/test_litellm/proxy/test_prometheus_cleanup.py
View file

@ -249,48 +249,26 @@ class TestMaybeSetupPrometheusMultiprocDir:
remaining = glob.glob(os.path.join(custom_dir, "*.db"))
assert remaining == []
def test_noop_for_single_worker(self):
"""Single worker doesn't need multiproc dir."""
litellm_settings = {"callbacks": ["prometheus"]}
@pytest.mark.parametrize(
"num_workers, litellm_settings",
[
(1, {"callbacks": ["prometheus"]}),
(4, {"callbacks": ["langfuse"]}),
(4, None),
],
)
def test_noop_when_setup_not_needed(self, num_workers, litellm_settings):
with patch.dict(os.environ, {}, clear=False):
os.environ.pop("PROMETHEUS_MULTIPROC_DIR", None)
os.environ.pop("prometheus_multiproc_dir", None)
ProxyInitializationHelpers._maybe_setup_prometheus_multiproc_dir(
num_workers=1,
num_workers=num_workers,
litellm_settings=litellm_settings,
)
assert os.environ.get("PROMETHEUS_MULTIPROC_DIR") is None
def test_noop_without_prometheus_callback(self):
"""No prometheus callback = no setup needed."""
litellm_settings = {"callbacks": ["langfuse"]}
with patch.dict(os.environ, {}, clear=False):
os.environ.pop("PROMETHEUS_MULTIPROC_DIR", None)
os.environ.pop("prometheus_multiproc_dir", None)
ProxyInitializationHelpers._maybe_setup_prometheus_multiproc_dir(
num_workers=4,
litellm_settings=litellm_settings,
)
assert os.environ.get("PROMETHEUS_MULTIPROC_DIR") is None
def test_noop_with_none_litellm_settings(self):
"""None litellm_settings = no setup needed."""
with patch.dict(os.environ, {}, clear=False):
os.environ.pop("PROMETHEUS_MULTIPROC_DIR", None)
ProxyInitializationHelpers._maybe_setup_prometheus_multiproc_dir(
num_workers=4,
litellm_settings=None,
)
assert os.environ.get("PROMETHEUS_MULTIPROC_DIR") is None
def test_prometheus_in_success_callback(self):
"""Prometheus in success_callback should also trigger setup."""
litellm_settings = {"success_callback": ["prometheus"]}