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