def test_add_entities_no_dups(self):
doc = self.doc
e = Entity()
e.group = 'group'
e.name = u'name'
de = DocumentEntity()
de.entity = e
de.relevance = 1.0
de.count = 2
doc.add_entity(de)
self.assertEqual([de], list(doc.entities))
e2 = Entity()
e2.group = 'group'
e2.name = u'name'
de2 = DocumentEntity()
de2.entity = e
de2.relevance = 0.5
de2.count = 3
doc.add_entity(de2)
# shouldn't add dup
self.assertEqual([de], list(doc.entities))
def test_guess_genders(self):
d = Document()
d.text = 'Fred Astair did something. He also did something else.'
de = DocumentEntity()
de.document = d
de.entity = Entity.query.get(self.fx.EntityData.sue_no_gender.id)
de.offset_list = '27:2'
self.db.session.add(d)
self.ex.guess_genders(d)
self.assertEqual('Male', d.entities[0].entity.person.gender.name)
def test_add_entities_no_dups(self):
doc = Document()
e = Entity()
e.group = 'group'
e.name = 'name'
de = DocumentEntity()
de.entity = e
de.relevance = 1.0
de.count = 2
doc.add_entity(de)
self.assertEqual([de], list(doc.entities))
e2 = Entity()
e2.group = 'group'
e2.name = 'name'
de2 = DocumentEntity()
de2.entity = e
de2.relevance = 0.5
de2.count = 3
doc.add_entity(de2)
# shouldn't add dup
self.assertEqual([de], list(doc.entities))
def test_delete_document(self):
doc = Document()
doc.text = 'And Fred said "Hello" to everyone.'
doc.published_at = datetime.datetime.utcnow()
u = Utterance()
u.entity = Entity()
u.entity.group = 'person'
u.entity.name = 'Fred'
u.quote = 'Hello'
self.assertTrue(doc.add_utterance(u))
de = DocumentEntity()
de.document = doc
de.entity = Entity.query.first()
de.relevance = 0.5
self.db.session.add(doc)
self.db.session.commit()
self.db.session.delete(doc)
self.db.session.commit()
def test_delete_document(self):
doc = self.doc
doc.text = u'And Fred said "Hello" to everyone.'
doc.published_at = datetime.datetime.utcnow()
u = Utterance()
u.entity = Entity()
u.entity.group = 'person'
u.entity.name = u'Fred'
u.quote = u'Hello'
self.assertTrue(doc.add_utterance(u))
de = DocumentEntity()
de.document = doc
de.relevance = 0.5
de.entity = Entity.query.first()
self.db.session.add(doc)
self.db.session.commit()
self.db.session.delete(doc)
self.db.session.commit()
def test_guess_genders(self):
d = Document.query.get(self.fx.DocumentData.simple.id)
d.text = 'Fred Astair did something. He also did something else.'
de = DocumentEntity()
de.document = d
de.relevance = 1.0
de.entity = Entity.query.get(self.fx.EntityData.sue_no_gender.id)
de.offset_list = '27:2'
self.db.session.add(d)
self.ex.guess_genders(d)
self.assertEqual('Male', d.entities[0].entity.person.gender.name)
def test_offsets_empty(self):
de = DocumentEntity()
self.assertEqual([], de.offsets())
def test_offsets_add_exists(self):
de = DocumentEntity()
de.offset_list = '1:2 3:4'
self.assertFalse(de.add_offset((3, 4)))
self.assertEqual('1:2 3:4', de.offset_list)
def test_offsets_add_new(self):
de = DocumentEntity()
de.offset_list = '1:2 3:4'
self.assertTrue(de.add_offset((4, 5)))
self.assertEqual('1:2 3:4 4:5', de.offset_list)
def test_offsets(self):
de = DocumentEntity()
de.offset_list = ' 4:5 1:2 3:4 '
self.assertEqual([(1,2), (3, 4), (4, 5)], de.offsets())
def test_offsets_empty(self):
de = DocumentEntity()
self.assertEqual([], de.offsets())
def test_offsets_add_exists(self):
de = DocumentEntity()
de.offset_list = '1:2 3:4'
self.assertFalse(de.add_offset((3, 4)))
self.assertEqual('1:2 3:4', de.offset_list)
def test_offsets_add_new(self):
de = DocumentEntity()
de.offset_list = '1:2 3:4'
self.assertTrue(de.add_offset((4, 5)))
self.assertEqual('1:2 3:4 4:5', de.offset_list)
def test_offsets(self):
de = DocumentEntity()
de.offset_list = ' 4:5 1:2 3:4 '
self.assertEqual([(1, 2), (3, 4), (4, 5)], de.offsets())
