def test_entities(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.add_entity_to_context([{ "key": "existing_key", "type": "existing_key", "source": "detection", "entity_message_index": 0 }], "include", 80.0, "new_key", "new_key") self.assertEqual([{ 'entity_message_index': 0, 'key': 'existing_key', 'source': 'detection', 'type': 'existing_key' }, { 'confidence': 80.0, 'entity_message_index': 1, 'key': 'new_key', 'source': 'detection', 'type': 'new_key', 'type_weighting': 30.0 }], actual)
def test_existing_entity(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.add_entity_to_context( [ { "key": "existing_key", "confidence": 80.0, "type": "existing_key", "source": "detection", "entity_message_index": 0, "type_weighting": 30.0, } ], "include", 80.0, "existing_key", "existing_key", ) self.assertEqual( [ { "entity_message_index": 1, "key": "existing_key", "confidence": 80.0, "source": "detection", "type": "existing_key", "type_weighting": 33.0, } ], actual, )
def test_outcome_exclude_entities(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.add_entity_to_context( [{"key": "existing_key", "type": "existing_key", "source": "detection", "entity_message_index": 0}], "exclude", 80.0, "new_key", "new_type", ) self.assertEqual( [ {"entity_message_index": 0, "key": "existing_key", "source": "detection", "type": "existing_key"}, { "confidence": 80.0, "entity_message_index": 1, "negation_modifier": -100, "key": "new_type", "source": "detection", "type": "new_key", "type_weighting": 30.0, }, ], actual, )
def test_low_confidence_entities(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.add_entity_to_context( [{"key": "existing_key", "type": "existing_key", "source": "detection", "entity_message_index": 0}], "include", 40.0, "new_key", "new_key", ) self.assertEqual( [{"entity_message_index": 0, "key": "existing_key", "source": "detection", "type": "existing_key"}], actual )
def test_existing_exclude_entity_now_include(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.add_entity_to_context([{ 'confidence': 80.0, 'entity_message_index': 1, 'negation_modifier': -100, 'key': 'black', 'source': 'detection', 'type': 'color', 'type_weighting': 90.0 }], "include", 80.0, "color", "black") self.assertEqual([{ 'confidence': 80.0, 'entity_message_index': 2, 'key': 'black', 'source': 'detection', 'type': 'color', 'type_weighting': 99.00000000000001 }], actual)
def test_regular(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) target.add_entity_to_context = MagicMock( side_effect=(["context_entities_1"])) actual = target.update_entities_with_last_message( ["entity_value"], { "detection": { "outcomes": [{ "intent": "intent_value_1", "entities": [{ "confidence": "confidence_value_1_1", "type": "type_value_1_1", "key": "key_value_1_1", "source": "detection" }] }] } }) self.assertEqual(1, target.add_entity_to_context.call_count) self.assertEqual(['entity_value'], target.add_entity_to_context.call_args_list[0][0][0]) self.assertEqual('intent_value_1', target.add_entity_to_context.call_args_list[0][0][1]) self.assertEqual('confidence_value_1_1', target.add_entity_to_context.call_args_list[0][0][2]) self.assertEqual('type_value_1_1', target.add_entity_to_context.call_args_list[0][0][3]) self.assertEqual('key_value_1_1', target.add_entity_to_context.call_args_list[0][0][4]) self.assertEqual("context_entities_1", actual)
def test_existing_exclude_entity(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.add_entity_to_context( [ { "confidence": 80.0, "entity_message_index": 1, "negation_modifier": -100, "key": "black", "source": "detection", "type": "color", "type_weighting": 90.0, } ], "exclude", 80.0, "color", "black", ) self.assertEqual( [ { "confidence": 80.0, "entity_message_index": 2, "negation_modifier": -100, "key": "black", "source": "detection", "type": "color", "type_weighting": 99.00000000000001, } ], actual, )
def test_regular(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) target.add_entity_to_context = MagicMock(side_effect=(["context_entities_1"])) actual = target.update_entities_with_last_message( ["entity_value"], { "detection": { "outcomes": [ { "intent": "intent_value_1", "entities": [ { "confidence": "confidence_value_1_1", "type": "type_value_1_1", "key": "key_value_1_1", "source": "detection", } ], } ] } }, ) self.assertEqual(1, target.add_entity_to_context.call_count) self.assertEqual(["entity_value"], target.add_entity_to_context.call_args_list[0][0][0]) self.assertEqual("intent_value_1", target.add_entity_to_context.call_args_list[0][0][1]) self.assertEqual("confidence_value_1_1", target.add_entity_to_context.call_args_list[0][0][2]) self.assertEqual("type_value_1_1", target.add_entity_to_context.call_args_list[0][0][3]) self.assertEqual("key_value_1_1", target.add_entity_to_context.call_args_list[0][0][4]) self.assertEqual("context_entities_1", actual)