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_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_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_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())
