phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add unit test

Browse source
This commit is contained in:
freshman829 2025-04-06 15:02:00 -05:00
parent c62b65a9d7
commit dd79517f2d
1 changed files with 24 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/litellm/test_init.py Normal file
View file

@ -0,0 +1,24 @@
# tests/litellm/test_init.py
import importlib
import sys
def test_api_key_from_env(monkeypatch):
monkeypatch.setenv("LITELLM_API_KEY", "mocked_key")
if "litellm" in sys.modules:
del sys.modules["litellm"]
import litellm
importlib.reload(litellm)
assert litellm.api_key == "mocked_key"
def test_api_base_from_env(monkeypatch):
monkeypatch.setenv("LITELLM_API_BASE", "https://mocked-base.com")
if "litellm" in sys.modules:
del sys.modules["litellm"]
import litellm
importlib.reload(litellm)
assert litellm.api_base == "https://mocked-base.com"