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)
