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
29 lines
680 B
Python
29 lines
680 B
Python
import pytest
|
|
import unittest
|
|
|
|
from promptflow.connections import CustomConnection
|
|
from my_tool_package.tools.my_tool_1 import my_tool
|
|
|
|
|
|
@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
|
|
|
|
|
|
class TestMyTool1:
|
|
def test_my_tool_1(self, my_custom_connection):
|
|
result = my_tool(my_custom_connection, input_text="Microsoft")
|
|
assert result == "Hello Microsoft"
|
|
|
|
|
|
# Run the unit tests
|
|
if __name__ == "__main__":
|
|
unittest.main()
|