def test_empty_entities(self):
context_data = MagicMock()
attribute_product_data = MagicMock()
target = Target(context_data, attribute_product_data)
actual = target.extract_entities({"entities": []})
self.assertListEqual([], actual)
def test_none_context(self):
context_data = MagicMock()
attribute_product_data = MagicMock()
target = Target(context_data, attribute_product_data)
actual = target.extract_entities(None)
self.assertListEqual([], 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_entities(self):
context_data = MagicMock()
attribute_product_data = MagicMock()
target = Target(context_data, attribute_product_data)
actual = target.extract_entities({"entities": ["entity_1", "entity_2"]})
self.assertListEqual(["entity_1", "entity_2"], actual)
def test_none_context(self):
context_data = MagicMock()
attribute_product_data = MagicMock()
target = Target(context_data, attribute_product_data)
actual = target.extract_entities(None)
self.assertListEqual([], 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')
