diff --git a/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java b/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java index fee09aaa5f22..2bd39208fbfa 100644 --- a/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java +++ b/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java @@ -136,20 +136,20 @@ class SkeletonTestTask extends PyExecutionFixtureTestTask { StreamUtil.readText(new FileInputStream(new File(getTestDataPath(), myExpectedSkeletonFile)), Charset.defaultCharset()); - Assert.assertThat("Wrong skeleton generated", removeGeneratorVersion(actual), - Matchers.equalToIgnoringCase(removeGeneratorVersion(skeletonText))); + Assert.assertThat("Wrong skeleton generated", removeComments(actual), + Matchers.equalToIgnoringCase(removeComments(skeletonText))); } myFixture.configureByFile(skeletonFile.getName()); } /** - * Removes strings that starts with "#", because generator version and other stuff may change + * Removes strings that starts with "#" and """, because generator version and other stuff may change * * @param textToClean text to remove strings from * @return text after cleanup */ - private static String removeGeneratorVersion(@NotNull final String textToClean) { - return textToClean.replaceAll("#.+", ""); + private static String removeComments(@NotNull final String textToClean) { + return textToClean.replaceAll("(#|\"\"\").+", ""); }