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