def test_add_classification_for_language_classify(self):
"""
Test add_classification_for_language function, use classify function to test
"""
ndb.get_context().clear_cache()
data = {"statements": ["for", "if", "else"], "operators": ['signed'], "data_types": ["int", "long", "float", "char",
"double"], "expressions": ['sizeof'], "other": []}
tr = TranslationEngine("C")
ce = ContributionEngine("C")
ce.add_classification_for_language(data)
self.assertEqual("statement", tr.classify_keywords("for", "keyword"))
self.assertEqual("data type", tr.classify_keywords("int", "keyword"))
self.assertEqual("expression", tr.classify_keywords("sizeof", "keyword"))
self.assertEqual("operator", tr.classify_keywords("signed", "keyword"))
