wehub-resource-sync
e768098d0e
tools_continuous_delivery / Private PyPI non-main branch release (push) Has been skipped
tools_continuous_delivery / Private PyPI main branch release (push) Failing after 2m42s
Publish Promptflow Doc / Build (push) Has been cancelled
Publish Promptflow Doc / Deploy (push) Has been cancelled
Flake8 Lint / flake8 (push) Has been cancelled
Spell check CI / Spell_Check (push) Has been cancelled
35 lines
830 B
Python
35 lines
830 B
Python
import pytest
|
|
import unittest
|
|
|
|
from promptflow.connections import CustomConnection
|
|
from my_tool_package.tools.my_tool_2 import MyTool
|
|
|
|
|
|
@pytest.fixture
|
|
def my_custom_connection() -> CustomConnection:
|
|
my_custom_connection = CustomConnection(
|
|
{
|
|
"api-key" : "my-api-key",
|
|
"api-secret" : "my-api-secret",
|
|
"api-url" : "my-api-url"
|
|
}
|
|
)
|
|
return my_custom_connection
|
|
|
|
|
|
@pytest.fixture
|
|
def my_tool_provider(my_custom_connection) -> MyTool:
|
|
my_tool_provider = MyTool(my_custom_connection)
|
|
return my_tool_provider
|
|
|
|
|
|
class TestMyTool2:
|
|
def test_my_tool_2(self, my_tool_provider: MyTool):
|
|
result = my_tool_provider.my_tool(input_text="Microsoft")
|
|
assert result == "Hello Microsoft"
|
|
|
|
|
|
# Run the unit tests
|
|
if __name__ == "__main__":
|
|
unittest.main()
|