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_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_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_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)