Fix isRecursiveObject

tamarinvs19 · tamarinvs19 · commit 4374a26c318e · 2023-11-29T11:07:46.000+03:00
diff --git a/utbot-python/src/main/kotlin/org/utbot/python/framework/api/python/PythonTree.kt b/utbot-python/src/main/kotlin/org/utbot/python/framework/api/python/PythonTree.kt
@@ -28,14 +28,13 @@ object PythonTree {
         return isRecursiveObjectDFS(tree, mutableSetOf())
     }
 
-    private fun isRecursiveObjectDFS(tree: PythonTreeNode, visited: MutableSet<PythonTreeNode>): Boolean {
+    private fun isRecursiveObjectDFS(tree: PythonTreeNode, visited: Set<PythonTreeNode>): Boolean {
         if (tree is PrimitiveNode) {
             return false
         }
         if (visited.contains(tree))
             return true
-        visited.add(tree)
-        return tree.children.any { isRecursiveObjectDFS(it, visited) }
+        return tree.children.any { isRecursiveObjectDFS(it, visited + tree) }
     }
 
     fun containsFakeNode(tree: PythonTreeNode): Boolean {

Original file line number	Diff line number	Diff line change
`@@ -28,14 +28,13 @@ object PythonTree {`
`28`	`28`	`return isRecursiveObjectDFS(tree, mutableSetOf())`
`29`	`29`	`}`
`30`	`30`
`31`		`- private fun isRecursiveObjectDFS(tree: PythonTreeNode, visited: MutableSet<PythonTreeNode>): Boolean {`
	`31`	`+ private fun isRecursiveObjectDFS(tree: PythonTreeNode, visited: Set<PythonTreeNode>): Boolean {`
`32`	`32`	`if (tree is PrimitiveNode) {`
`33`	`33`	`return false`
`34`	`34`	`}`
`35`	`35`	`if (visited.contains(tree))`
`36`	`36`	`return true`
`37`		`- visited.add(tree)`
`38`		`- return tree.children.any { isRecursiveObjectDFS(it, visited) }`
	`37`	`+ return tree.children.any { isRecursiveObjectDFS(it, visited + tree) }`
`39`	`38`	`}`
`40`	`39`
`41`	`40`	`fun containsFakeNode(tree: PythonTreeNode): Boolean {`