blob: fd5960d0f079ec0610c0be41181ebae0fdb7085c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
# pylint: disable=no-self-use
"""Tests for the tool backend."""
from collections import Counter
import pytest
from aiet.backend.common import ConfigurationException
from aiet.backend.config import ToolConfig
from aiet.backend.tool import get_available_tool_directory_names
from aiet.backend.tool import get_available_tools
from aiet.backend.tool import get_tool
from aiet.backend.tool import Tool
def test_get_available_tool_directory_names() -> None:
"""Test get_available_tools mocking get_resources."""
directory_names = get_available_tool_directory_names()
assert Counter(directory_names) == Counter(["tool1", "tool2", "vela"])
def test_get_available_tools() -> None:
"""Test get_available_tools mocking get_resources."""
available_tools = get_available_tools()
expected_tool_names = sorted(
[
"tool_1",
"tool_2",
"vela",
"vela",
"vela",
]
)
assert all(isinstance(s, Tool) for s in available_tools)
assert all(s != 42 for s in available_tools)
assert any(s == available_tools[0] for s in available_tools)
assert len(available_tools) == len(expected_tool_names)
available_tool_names = sorted(str(s) for s in available_tools)
assert available_tool_names == expected_tool_names
def test_get_tool() -> None:
"""Test get_tool mocking get_resoures."""
tools = get_tool("tool_1")
assert len(tools) == 1
tool = tools[0]
assert tool is not None
assert isinstance(tool, Tool)
assert tool.name == "tool_1"
tools = get_tool("unknown tool")
assert not tools
def test_tool_creation() -> None:
"""Test edge cases when creating a Tool instance."""
with pytest.raises(ConfigurationException):
Tool(ToolConfig(name="test", commands={"test": []})) # no 'run' command
|
