def test_should_get_supported_builtin_entities(self):
        # When
        supported_entities = get_supported_entities("en")

        # Then
        self.assertIn("snips/number", supported_entities)
        self.assertIn("snips/datetime", supported_entities)
        for builtin in supported_entities:
            self.assertIsInstance(builtin, str)
Example #2
0
 def fit(self, dataset, intent):
     self.language = dataset[LANGUAGE]
     self.builtin_entities = list(get_supported_entities(self.language))
     self.args["entity_labels"] = self.builtin_entities
Example #3
0
 def fit(self, dataset, intent):
     self.language = dataset[LANGUAGE]
     self.builtin_entities = list(get_supported_entities(self.language))
     self.args["entity_labels"] = self.builtin_entities
 def test_should_get_builtin_entity_examples(self):
     for language in get_all_languages():
         for builtin_entity in get_supported_entities(language):
             examples = get_builtin_entity_examples(builtin_entity,
                                                    language)
             self.assertGreaterEqual(len(examples), 1)