mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-14 00:18:17 +08:00
7753ba2d37
Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Yeuoly <admin@srmxy.cn> Co-authored-by: JzoNg <jzongcode@gmail.com> Co-authored-by: StyleZhang <jasonapring2015@outlook.com> Co-authored-by: jyong <jyong@dify.ai> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: jyong <718720800@qq.com>
21 lines
672 B
Python
21 lines
672 B
Python
import pytest
|
|
from core.tools.tool_manager import ToolManager
|
|
|
|
provider_generator = ToolManager.list_builtin_providers()
|
|
provider_names = [provider.identity.name for provider in provider_generator]
|
|
ToolManager.clear_builtin_providers_cache()
|
|
provider_generator = ToolManager.list_builtin_providers()
|
|
|
|
@pytest.mark.parametrize('name', provider_names)
|
|
def test_tool_providers(benchmark, name):
|
|
"""
|
|
Test that all tool providers can be loaded
|
|
"""
|
|
|
|
def test(generator):
|
|
try:
|
|
return next(generator)
|
|
except StopIteration:
|
|
return None
|
|
|
|
benchmark.pedantic(test, args=(provider_generator,), iterations=1, rounds=1) |