public void FindAssignStatementsTest()
{
var methodBody = GenerateMethodBody(PythonCodeGenerator.GenerateDefaultAssignmentStatement());
var finder = BuildFinder(methodBody);
var assignStatements = finder.FindAssignStatements();
Assert.AreEqual(1, assignStatements.Count());
}
private string GenerateMethodBody(int statementsCount)
{
var statements = new List <string>();
for (var i = 0; i < statementsCount; i++)
{
statements.Add(PythonCodeGenerator.GenerateDefaultAssignmentStatement());
}
return(PythonCodeGenerator.GenerateFunctionBody(statements));
}
public void CountOfDirectStatementsInBodyTest()
{
var assignmentStatement = PythonCodeGenerator.GenerateDefaultAssignmentStatement();
var embeddedStatement = PythonCodeGenerator.GenerateDefaultCompoundStatement(2);
var returnStatement = PythonCodeGenerator.GenerateDefaultReturnStatement();
string[] statements = { assignmentStatement, embeddedStatement, returnStatement };
var methodBody = GenerateMethodBody(statements);
var finder = BuildFinder(methodBody);
var statementsCount = finder.CountOfDirectStatementsInBody();
Assert.AreEqual(3, statementsCount);
}