From 0faff91c722ca70b60fc4f51c2d732df498c7cd9 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Wed, 27 Nov 2024 06:52:31 -0800
Subject: [PATCH] test: enhance Python test case with class and type
 annotations

---
 tests/basic/test_repomap.py | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/tests/basic/test_repomap.py b/tests/basic/test_repomap.py
index 88f6cf89d..746c12d01 100644
--- a/tests/basic/test_repomap.py
+++ b/tests/basic/test_repomap.py
@@ -480,8 +480,33 @@ let () =
             ),
             "python": (
                 "test.py",
-                'def greet(name):\n    print(f"Hello, {name}!")\n',
-                "greet",  # Key symbol to check
+                '''from typing import Optional, List
+
+class Person:
+    """A class representing a person."""
+    
+    def __init__(self, name: str, age: Optional[int] = None):
+        self.name = name
+        self.age = age
+    
+    def greet(self, formal: bool = False) -> str:
+        """Generate a greeting."""
+        prefix = "Good day" if formal else "Hello"
+        return f"{prefix}, {self.name}!"
+
+def create_greeting_list(people: List[Person]) -> List[str]:
+    """Create greetings for a list of people."""
+    return [person.greet() for person in people]
+
+# Constants
+DEFAULT_NAME = "World"
+MAX_AGE = 150
+
+if __name__ == "__main__":
+    person = Person(DEFAULT_NAME)
+    print(person.greet())
+''',
+                "Person",  # Key symbol to check
             ),
             "ql": (
                 "test.ql",