def test_institution_facets(add_institution, add_citation): """ institution_facets() should provide a list of label/value/count dicts. """ i1 = add_institution(name='Institution 1') i2 = add_institution(name='Institution 2') i3 = add_institution(name='Institution 3') for i in range(3): c = add_citation() Institution_Document.create(institution=i1, document=c.document) for i in range(2): c = add_citation() Institution_Document.create(institution=i2, document=c.document) for i in range(1): c = add_citation() Institution_Document.create(institution=i3, document=c.document) Citation_Index.es_insert() Institution_Index.es_insert() facets = institution_facets() assert facets == [ dict(label='Institution 1', value=i1.id, count=3), dict(label='Institution 2', value=i2.id, count=2), dict(label='Institution 3', value=i3.id, count=1), ]
def test_merge_included(add_institution, add_citation): """ When ids are passed for institutions that fall outside of the default page, merge the extra facets into the baseline ranking. """ i1 = add_institution(name='Institution 1') i2 = add_institution(name='Institution 2') i3 = add_institution(name='Institution 3') for i in range(3): c = add_citation() Institution_Document.create(institution=i1, document=c.document) for i in range(2): c = add_citation() Institution_Document.create(institution=i2, document=c.document) for i in range(1): c = add_citation() Institution_Document.create(institution=i3, document=c.document) Citation_Index.es_insert() Institution_Index.es_insert() facets = institution_facets(depth=2, include=[i2.id, i3.id]) assert facets == [ dict(label='Institution 1', value=i1.id, count=3), dict(label='Institution 2', value=i2.id, count=2), # Dedupe 2. dict(label='Institution 3', value=i3.id, count=1), # Append 3. ]
def test_institution_facets(add_institution, add_citation): """ institution_facets() should provide a list of label/value/count dicts. """ i1 = add_institution(name='Institution 1') i2 = add_institution(name='Institution 2') i3 = add_institution(name='Institution 3') for i in range(3): c = add_citation() Institution_Document.create(institution=i1, document=c.document) for i in range(2): c = add_citation() Institution_Document.create(institution=i2, document=c.document) for i in range(1): c = add_citation() Institution_Document.create(institution=i3, document=c.document) Citation_Index.es_insert() Institution_Index.es_insert() facets = institution_facets() assert facets == [ dict(label='Institution 1', value=i1.id, count=3), dict(label='Institution 2', value=i2.id, count=2), dict(label='Institution 3', value=i3.id, count=1), ]
def test_es_insert(es, add_institution): """ Institution_Index.es_insert() should index institutions. """ for i in range(10): add_institution('inst' + str(i)) Institution_Index.es_insert() for inst in Institution.select(): doc = config.es.get( index='institution', id=inst.id, ) assert doc['_source']['name'] == inst.name
def test_es_insert(es, add_institution): """ Institution_Index.es_insert() should index institutions. """ for i in range(10): add_institution('inst'+str(i)) Institution_Index.es_insert() for inst in Institution.select(): doc = config.es.get( index='institution', id=inst.id, ) assert doc['_source']['name'] == inst.name