wehub-resource-sync
35 lines
998 B
Python
35 lines
998 B
Python
from demo_tool_agent import TimePlugin, CalculatorPlugin
|
|
|
|
def test_calculator_add():
|
|
calc = CalculatorPlugin()
|
|
result = calc.add(5, 7)
|
|
assert result == 12
|
|
print("test_calculator_add passed")
|
|
|
|
def test_calculator_subtract():
|
|
calc = CalculatorPlugin()
|
|
result = calc.subtract(10, 4)
|
|
assert result == 6
|
|
print("test_calculator_subtract passed")
|
|
|
|
def test_time_plugin():
|
|
# We can't easily test exact time, but we can check format
|
|
time_plugin = TimePlugin()
|
|
time_str = time_plugin.get_current_time()
|
|
# Expect format YYYY-MM-DD HH:MM:SS
|
|
assert len(time_str) == 19
|
|
assert "-" in time_str
|
|
assert ":" in time_str
|
|
print("test_time_plugin passed")
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
test_calculator_add()
|
|
test_calculator_subtract()
|
|
test_time_plugin()
|
|
print("All tests passed!")
|
|
except AssertionError as e:
|
|
print(f"Test failed: {e}")
|
|
except Exception as e:
|
|
print(f"An error occurred: {e}")
|