def test_java_snippet1(self): (is_snippet, _) = je.is_java_snippet(self.snippet1, self.filters) self.assertTrue(is_snippet) (is_snippet, _) = je.is_java_snippet(self.snippet2, self.filters) self.assertFalse(is_snippet) (is_snippet, _) = je.is_java_snippet(self.snippet3, self.filters) self.assertTrue(is_snippet)
def classify_code_snippet(text, filters): code = None try: if is_xml_snippet(text)[0]: language = XML_LANGUAGE elif is_java_snippet(text, filters[JAVA_LANGUAGE])[0]: language = JAVA_LANGUAGE else: language = OTHER_LANGUAGE code = CodeSnippet( language=language, snippet_text=text, ) code.save() except Exception: logger.exception('Error while classifying snippet.') return code