def test_extract_single_entity(self):
entity_value = 'kanyewest'
data = {
'entities': {
'hashtags': [{'text': entity_value}]
}
}
actual_output = utils.extract_entity(data, 'entities.hashtags', 'text')
self.assertEqual(entity_value, actual_output)
def test_extract_multiple_entity(self):
value_1 = 'kanyewest'
value_2 = 'dannybrown'
data = {
'entities': {
'hashtags': [{'text': value_1}, {'text': value_2}]
}
}
expected_output = '{}/{}'.format(value_1, value_2)
actual_output = utils.extract_entity(data, 'entities.hashtags', 'text')
self.assertEqual(expected_output, actual_output)
def test_extract_multiple_entity(self):
value_1 = 'kanyewest'
value_2 = 'dannybrown'
data = {
'entities': {
'hashtags': [{
'text': value_1
}, {
'text': value_2
}]
}
}
expected_output = '{}/{}'.format(value_1, value_2)
actual_output = utils.extract_entity(data, 'entities.hashtags', 'text')
self.assertEqual(expected_output, actual_output)
def test_extract_single_entity(self):
entity_value = 'kanyewest'
data = {'entities': {'hashtags': [{'text': entity_value}]}}
actual_output = utils.extract_entity(data, 'entities.hashtags', 'text')
self.assertEqual(entity_value, actual_output)
