test(tools): add unit test for using class method as tool

Author: Ponprabhakar-pg

tests-integ/test_method_tools.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+"""
+Test script for class-method-based tools
+"""
+
+import logging
+from typing import Optional
+
+from strands import Agent, tool
+
+logging.getLogger("strands").setLevel(logging.DEBUG)
+logging.basicConfig(format="%(levelname)s | %(name)s | %(message)s", handlers=[logging.StreamHandler()])
+
+
+class TestAgent:
+    def __init__(self):
+        self.agent = None
+
+    @tool
+    def word_counter(self, text: str) -> str:
+        """
+        Count words in text.
+
+        Args:
+            text: Text to analyze
+        """
+        count = len(text.split())
+        return f"Word count: {count}"
+
+    @tool(name="count_chars", description="Count characters in text")
+    def count_chars(self, text: str, include_spaces: Optional[bool] = True) -> str:
+        """
+        Count characters in text.
+
+        Args:
+            text: Text to analyze
+            include_spaces: Whether to include spaces in the count
+        """
+        if not include_spaces:
+            text = text.replace(" ", "")
+        return f"Character count: {len(text)}"
+
+    def initialize_agent(self, tools=None) -> None:
+        self.agent = Agent(tools=tools or [])
+
+    def get_agent(self) -> Agent:
+        if not self.agent:
+            self.agent = Agent()
+        return self.agent
+
+
+# Object test agent class
+test_agent = TestAgent()
+# Initialize agent where tools are defined as class methods
+test_agent.initialize_agent([test_agent.word_counter, test_agent.count_chars])
+# Get the initialized agent
+agent = test_agent.get_agent()
+
+print("\n===== Testing Direct Tool Access =====")
+# Use the tools directly
+word_result = agent.tool.word_counter(text="Hello world, this is a test")
+print(f"\nWord counter result: {word_result}")
+
+char_result = agent.tool.count_chars(text="Hello world!", include_spaces=False)
+print(f"\nCharacter counter result: {char_result}")
+
+print("\n===== Testing Natural Language Access =====")
+# Use through natural language
+nl_result = agent("Count the words in this sentence: 'The quick brown fox jumps over the lazy dog'")
+print(f"\nNL Result: {nl_result}")