Spaces:

DesertWolf
/

test3

Paused

test3 / tests /test_litellm /proxy /test_proxy_utils.py

Upload folder using huggingface_hub

447ebeb verified over 1 year ago

518 Bytes

import json
import os
import sys

import pytest
from fastapi.testclient import TestClient

sys.path.insert(
    0, os.path.abspath("../../..")
)  # Adds the parent directory to the system path


from unittest.mock import MagicMock

from litellm.proxy.utils import get_custom_url


def test_get_custom_url(monkeypatch):
    monkeypatch.setenv("SERVER_ROOT_PATH", "/litellm")
    custom_url = get_custom_url(request_base_url="http://0.0.0.0:4000", route="ui/")
    assert custom_url == "http://0.0.0.0:4000/litellm/ui/"

	import json
	import os
	import sys

	import pytest
	from fastapi.testclient import TestClient

	sys.path.insert(
	0, os.path.abspath("../../..")
	) # Adds the parent directory to the system path


	from unittest.mock import MagicMock

	from litellm.proxy.utils import get_custom_url


	def test_get_custom_url(monkeypatch):
	monkeypatch.setenv("SERVER_ROOT_PATH", "/litellm")
	custom_url = get_custom_url(request_base_url="http://0.0.0.0:4000", route="ui/")
	assert custom_url == "http://0.0.0.0:4000/litellm/ui/"