public void FindVariablesInReturnStatement() { var variableId = "variable1"; var returnStatement = PythonCodeGenerator.GenerateReturnStatement(variableId, 0); var finder = BuildFinder(returnStatement); var variables = finder.FindVariables(); Assert.AreEqual(1, variables.Count()); }