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(self): de = DocumentEntity() de.offset_list = ' 4:5 1:2 3:4 ' self.assertEqual([(1, 2), (3, 4), (4, 5)], de.offsets())