tests galore

Author: Codeflash Bot

code_to_optimize/topological_sort.py

@@ -28,4 +28,4 @@ def topologicalSort(self):
             if visited[i] == False:
                 self.topologicalSortUtil(i, visited, stack)
 
-        return stack, str(sorting_id)
+        return stack, str(sorting_id)

codeflash/discovery/discover_unit_tests.py

@@ -307,6 +307,16 @@ def visit_ImportFrom(self, node: ast.ImportFrom) -> None:
                 self.found_qualified_name = qname
                 return
 
+            # Check if any target function is a method of the imported class/module
+            # e.g., importing Graph and looking for Graph.topologicalSort
+            for target_func in fnames:
+                if "." in target_func:
+                    class_name, method_name = target_func.split(".", 1)
+                    if aname == class_name:
+                        self.found_any_target_function = True
+                        self.found_qualified_name = target_func
+                        return
+
             prefix = qname + "."
             # Only bother if one of the targets startswith the prefix-root
             candidates = proots.get(qname, ())