Exemplo n.º 1
0
    def test_analyze_batch_actions_result_extract_key_phrases(
            self, extract_key_phrases_result):
        model = _models.AnalyzeBatchActionsResult(
            document_results=[extract_key_phrases_result[0]],
            is_error=False,
            action_type=_models.AnalyzeBatchActionsType.EXTRACT_KEY_PHRASES,
            completed_on=datetime.datetime(1, 1, 1))

        model_repr = (
            "AnalyzeBatchActionsResult(document_results=[{}], is_error={}, action_type={}, completed_on={})"
            .format(extract_key_phrases_result[1], False,
                    "extract_key_phrases", datetime.datetime(1, 1, 1)))

        assert repr(model) == model_repr
Exemplo n.º 2
0
    def test_analyze_batch_actions_result_recognize_pii_entities(
            self, recognize_pii_entities_result):
        model = _models.AnalyzeBatchActionsResult(
            document_results=[recognize_pii_entities_result[0]],
            is_error=False,
            action_type=_models.AnalyzeBatchActionsType.RECOGNIZE_PII_ENTITIES,
            completed_on=datetime.datetime(1, 1, 1))

        model_repr = (
            "AnalyzeBatchActionsResult(document_results=[{}], is_error={}, action_type={}, completed_on={})"
            .format(recognize_pii_entities_result[1], False,
                    "recognize_pii_entities", datetime.datetime(1, 1, 1)))

        assert repr(model) == model_repr