def test_single_entity(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.create_entity_type_index_modifier([{ "key": "color", "type": "black", "confidence": 100, "type_weighting": 50, "source": "detection", "entity_message_index": 0 }]) self.assertListEqual([{ 'confidence': 100, 'entity_message_index': 0, 'key': 'color', 'source': 'detection', 'type': 'black', 'type_index_modifier': 1.0, 'type_weighting': 50 }], actual)
def test_last_message_user(self): context_data = MagicMock() context_data.get = MagicMock(return_value="existing_context_value") attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) target.extract_last_user_message = MagicMock(return_value="last_message") target.extract_entities = MagicMock(return_value="extracted_entities") target.remove_default_entities_if_detections = MagicMock(return_value="removed_entities") target.update_entities_with_last_message = MagicMock(return_value="entities_update_entities_with_last_message") target.create_entity_type_index_modifier = MagicMock(return_value="entities_create_entity_type_index_modifier") target.change_entities_weighting = MagicMock(return_value="entities_change_entities_weighting") target.add_product_counts = MagicMock(return_value="entities_add_product_counts") target.split_unsupported_entities = MagicMock(return_value=("supported_entities", "unsupported_entities")) target.update("context_id_value", "_rev_value", "messages_value") target.extract_last_user_message.assert_called_once_with("messages_value") context_data.get.assert_called_once_with("context_id_value", "_rev_value") target.extract_entities.assert_called_once_with("existing_context_value") target.update_entities_with_last_message.assert_called_once_with("extracted_entities", "last_message") target.remove_default_entities_if_detections.assert_called_once_with( "entities_update_entities_with_last_message" ) target.create_entity_type_index_modifier.assert_called_once_with("removed_entities") target.change_entities_weighting.assert_called_once_with("entities_create_entity_type_index_modifier") target.add_product_counts.assert_called_once_with("entities_change_entities_weighting") target.split_unsupported_entities.assert_called_once_with("entities_add_product_counts") context_data.update.assert_called_once_with( "context_id_value", "_rev_value", unsupported_entities="unsupported_entities", entities="supported_entities" )
def test_single_entity(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.create_entity_type_index_modifier( [ { "key": "color", "type": "black", "confidence": 100, "type_weighting": 50, "source": "detection", "entity_message_index": 0, } ] ) self.assertListEqual( [ { "confidence": 100, "entity_message_index": 0, "key": "color", "source": "detection", "type": "black", "type_index_modifier": 1.0, "type_weighting": 50, } ], actual, )
def test_empty_entities(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.create_entity_type_index_modifier([]) self.assertListEqual([], actual)
def test_empty_entities(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.create_entity_type_index_modifier([]) self.assertListEqual([], actual)
def test_multiple_entity_same_type_different_message_index(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.create_entity_type_index_modifier([{ "type": "color", "key": "black", "confidence": 100, "type_weighting": 50, "source": "detection", "entity_message_index": 0 }, { "key": "red", "type": "color", "confidence": 100, "type_weighting": 50, "source": "detection", "entity_message_index": 1 }]) self.assertListEqual([{ 'confidence': 100, 'entity_message_index': 0, 'key': 'black', 'source': 'detection', 'type': 'color', 'type_index_modifier': 0.7071067811865475, 'type_weighting': 50 }, { 'confidence': 100, 'entity_message_index': 1, 'key': 'red', 'source': 'detection', 'type': 'color', 'type_index_modifier': 1.0, 'type_weighting': 50 }], actual)
def test_multiple_entity_same_type_different_message_index(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) actual = target.create_entity_type_index_modifier( [ { "type": "color", "key": "black", "confidence": 100, "type_weighting": 50, "source": "detection", "entity_message_index": 0, }, { "key": "red", "type": "color", "confidence": 100, "type_weighting": 50, "source": "detection", "entity_message_index": 1, }, ] ) self.assertListEqual( [ { "confidence": 100, "entity_message_index": 0, "key": "black", "source": "detection", "type": "color", "type_index_modifier": 0.7071067811865475, "type_weighting": 50, }, { "confidence": 100, "entity_message_index": 1, "key": "red", "source": "detection", "type": "color", "type_index_modifier": 1.0, "type_weighting": 50, }, ], actual, )
def test_last_message_user(self): context_data = MagicMock() context_data.get = MagicMock(return_value="existing_context_value") attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) target.extract_last_user_message = MagicMock( return_value="last_message") target.extract_entities = MagicMock(return_value="extracted_entities") target.remove_default_entities_if_detections = MagicMock( return_value="removed_entities") target.update_entities_with_last_message = MagicMock( return_value="entities_update_entities_with_last_message") target.create_entity_type_index_modifier = MagicMock( return_value="entities_create_entity_type_index_modifier") target.change_entities_weighting = MagicMock( return_value="entities_change_entities_weighting") target.add_product_counts = MagicMock( return_value="entities_add_product_counts") target.split_unsupported_entities = MagicMock( return_value=("supported_entities", "unsupported_entities")) target.update("context_id_value", "_rev_value", "messages_value") target.extract_last_user_message.assert_called_once_with( "messages_value") context_data.get.assert_called_once_with('context_id_value', '_rev_value') target.extract_entities.assert_called_once_with( 'existing_context_value') target.update_entities_with_last_message.assert_called_once_with( 'extracted_entities', 'last_message') target.remove_default_entities_if_detections.assert_called_once_with( 'entities_update_entities_with_last_message') target.create_entity_type_index_modifier.assert_called_once_with( 'removed_entities') target.change_entities_weighting.assert_called_once_with( 'entities_create_entity_type_index_modifier') target.add_product_counts.assert_called_once_with( 'entities_change_entities_weighting') target.split_unsupported_entities.assert_called_once_with( "entities_add_product_counts") context_data.update.assert_called_once_with( 'context_id_value', '_rev_value', unsupported_entities="unsupported_entities", entities='supported_entities')