public void ShouldDetectAllClassesInCode() { string[] classNames = { "class1", "class2" }; var code = PythonCodeGenerator.GenerateRoot(classNames); var parserBuilder = new ParserBuilder(); parserBuilder.Build(code); var finder = parserBuilder.GetParseElementFinder(); var classes = finder.FindClasses(); var actualDetectedClasses = classes.Count(); Assert.AreEqual(classNames.Length, actualDetectedClasses); }