[cfg] We don't visit functions in while loops

[KevinHock]

You can see we do it for for loops and not while loops

pyt/pyt/cfg/stmt_visitor.py

Lines 513 to 529 in 346a2d3

	if isinstance(node.iter, ast.Call) and get_call_names_as_string(node.iter.func) in self.function_names:
	last_node = self.visit(node.iter)
	last_node.connect(for_node)
	return self.loop_node_skeleton(for_node, node)
	def visit_While(self, node):
	label_visitor = LabelVisitor()
	label_visitor.visit(node.test)
	test = self.append_node(Node(
	'while ' + label_visitor.result + ':',
	node,
	path=self.filenames[-1]
	))
	return self.loop_node_skeleton(test, node)

[KevinHock]

I think reading the cfg readme is a good start on this

[KevinHock]

Hi @omergunal, would you like to take this one?

[omergunal]

i would 🎉