bibrank_selfcites_regression_tests.py
No OneTemporary
Actions

Subscribers

None

File Metadata

Created: Wed, Oct 2, 08:42

bibrank_selfcites_regression_tests.py
View Options

	# -- coding: utf-8 --
	##
	## This file is part of Invenio.
	## Copyright (C) 2011, 2012 CERN.
	##
	## Invenio is free software; you can redistribute it and/or
	## modify it under the terms of the GNU General Public License as
	## published by the Free Software Foundation; either version 2 of the
	## License, or (at your option) any later version.
	##
	## Invenio is distributed in the hope that it will be useful, but
	## WITHOUT ANY WARRANTY; without even the implied warranty of
	## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	## General Public License for more details.
	##
	## You should have received a copy of the GNU General Public License
	## along with Invenio; if not, write to the Free Software Foundation, Inc.,
	## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.

	"""Unit tests for the search engine query parsers."""


	import unittest
	import sys
	from StringIO import StringIO
	from datetime import datetime, timedelta

	from invenio.testutils import make_test_suite, run_test_suite


	class SelfCitesIndexerTests(unittest.TestCase):
	"""Test utility functions for the summarizer components"""

	def setUp(self):
	from invenio.bibrank_selfcites_task import fill_self_cites_tables
	fill_self_cites_tables({'algorithm': 'simple'})

	def test_get_personids_from_record(self):
	from invenio.bibrank_selfcites_indexer import get_personids_from_record
	get_personids_from_record(1)

	def test_get_authors_tags(self):
	"""test_get_authors_tags
	We don't care about the value since it's
	customizable but verify that it doesn't error
	"""
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	tags = get_authors_tags()
	self.assertEqual(len(tags), 4)

	def test_get_authors_from_record(self):
	from invenio.bibrank_selfcites_indexer import get_authors_from_record
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	from invenio.config import CFG_BIBRANK_SELFCITES_USE_BIBAUTHORID
	old_config = CFG_BIBRANK_SELFCITES_USE_BIBAUTHORID
	tags = get_authors_tags()
	CFG_BIBRANK_SELFCITES_USE_BIBAUTHORID = 0
	self.assert_(get_authors_from_record(1, tags))
	CFG_BIBRANK_SELFCITES_USE_BIBAUTHORID = 1
	get_authors_from_record(1, tags)
	CFG_BIBRANK_SELFCITES_USE_BIBAUTHORID = old_config

	def test_get_collaborations_from_record(self):
	from invenio.bibrank_selfcites_indexer import get_collaborations_from_record
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	tags = get_authors_tags()
	self.assert_(not get_collaborations_from_record(1, tags))

	def test_fetch_references(self):
	from invenio.bibrank_selfcites_indexer import fetch_references
	self.assertEqual(fetch_references(1), set())

	def test_get_precomputed_self_cites_list(self):
	from invenio.bibrank_selfcites_indexer import \
	get_precomputed_self_cites_list
	counts = get_precomputed_self_cites_list([1, 2, 3, 4])
	self.assertEqual(counts, ((1, 0), (2, 0), (3, 0), (4, 0)))

	def test_get_precomputed_self_cites(self):
	from invenio.bibrank_selfcites_indexer import \
	get_precomputed_self_cites
	ret = get_precomputed_self_cites(1)
	self.assertEqual(ret, 0)

	def test_compute_simple_self_citations(self):
	from invenio.bibrank_selfcites_indexer import \
	compute_simple_self_citations
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	tags = get_authors_tags()
	ret = compute_simple_self_citations(1, tags)
	self.assertEqual(ret, set())

	def test_compute_friends_self_citations(self):
	from invenio.bibrank_selfcites_indexer import \
	compute_friends_self_citations
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	tags = get_authors_tags()
	ret = compute_friends_self_citations(1, tags)
	self.assertEqual(ret, set())

	def test_get_self_citations_count(self):
	from invenio.bibrank_selfcites_indexer import get_self_citations_count
	ret = get_self_citations_count([1, 2, 3, 4])
	self.assertEqual(ret, 0)

	def test_update_self_cites_tables(self):
	from invenio.bibrank_selfcites_indexer import update_self_cites_tables
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	tags = get_authors_tags()
	config = {}
	update_self_cites_tables(1, config, tags)

	def test_store_record(self):
	from invenio.bibrank_selfcites_indexer import store_record
	from invenio.bibrank_selfcites_indexer import get_authors_from_record
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	from invenio.dbquery import run_sql
	tags = get_authors_tags()
	recid = 1
	authors = get_authors_from_record(recid, tags)
	sql = 'DELETE FROM rnkRECORDSCACHE WHERE id_bibrec = %s'
	run_sql(sql, (recid,))
	store_record(recid, authors)
	sql = 'SELECT count(*) FROM rnkRECORDSCACHE WHERE id_bibrec = %s'
	count = run_sql(sql, (recid,))[0][0]
	self.assert_(count)

	def test_get_author_coauthors_list(self):
	from invenio.bibrank_selfcites_indexer import get_author_coauthors_list
	from invenio.bibrank_selfcites_indexer import get_authors_from_record
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	tags = get_authors_tags()
	config = {'friends_threshold': 3}
	authors = get_authors_from_record(1, tags)
	self.assert_(get_author_coauthors_list(authors, config))

	def test_store_record_coauthors_with_some_deleted(self):
	from invenio.bibrank_selfcites_indexer import store_record_coauthors
	from invenio.bibrank_selfcites_indexer import get_authors_from_record
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	from invenio.dbquery import run_sql
	tags = get_authors_tags()
	config = {'friends_threshold': 3}
	recid = 1
	authors = get_authors_from_record(recid, tags)

	sql = 'DELETE FROM rnkEXTENDEDAUTHORS WHERE id = %s'
	run_sql(sql, (recid,))
	store_record_coauthors(recid, authors, [1], authors, config)
	sql = 'SELECT count(*) FROM rnkEXTENDEDAUTHORS WHERE id = %s'
	count = run_sql(sql, (recid,))[0][0]
	self.assert_(count)

	def test_store_record_coauthors_with_none_deleted(self):
	from invenio.bibrank_selfcites_indexer import store_record_coauthors
	from invenio.bibrank_selfcites_indexer import get_authors_from_record
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	from invenio.dbquery import run_sql
	tags = get_authors_tags()
	recid = 1
	config = {'friends_threshold': 3}
	authors = get_authors_from_record(recid, tags)

	sql = 'DELETE FROM rnkEXTENDEDAUTHORS WHERE id = %s'
	run_sql(sql, (recid,))
	store_record_coauthors(recid, authors, [], authors, config)
	sql = 'SELECT count(*) FROM rnkEXTENDEDAUTHORS WHERE id = %s'
	count = run_sql(sql, (recid,))[0][0]
	self.assert_(count)

	def test_get_record_coauthors(self):
	from invenio.bibrank_selfcites_indexer import get_record_coauthors
	self.assert_(get_record_coauthors(1))


	class SelfCitesTaskTests(unittest.TestCase):
	def test_check_options(self):
	from invenio.bibrank_selfcites_task import check_options
	old_stderr = sys.stderr
	sys.stderr = StringIO()
	try:
	self.assert_(not check_options())
	finally:
	sys.stderr = old_stderr

	def test_parse_option(self):
	from invenio.bibrank_selfcites_task import parse_option
	parse_option('-a', None, None, None)
	parse_option('-m', None, None, None)
	parse_option('-c', '1', None, None)
	parse_option('-r', '1', None, None)
	parse_option('--recids', '1-10', None, None)
	parse_option('-r', '1,2,3-6', None, None)
	parse_option('--rebuild', None, None, None)

	def test_compute_and_store_self_citations(self):
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	from invenio.bibrank_selfcites_task import compute_and_store_self_citations
	from invenio.bibrank_selfcites_task import get_citations_fun
	from invenio.bibrank_selfcites_indexer import ALL_ALGORITHMS

	tags = get_authors_tags()
	for algorithm in ALL_ALGORITHMS:
	citation_fun = get_citations_fun(algorithm=algorithm)
	compute_and_store_self_citations(1, tags, citation_fun)

	def test_rebuild_tables(self):
	from invenio.bibrank_selfcites_task import rebuild_tables
	from invenio.bibrank_selfcites_indexer import ALL_ALGORITHMS
	for algorithm in ALL_ALGORITHMS.iterkeys():
	config = {'algorithm': algorithm, 'friends_threshold': 3}
	assert rebuild_tables(config)

	def test_fetch_bibauthorid_last_update(self):
	from invenio.bibrank_selfcites_task import \
	fetch_bibauthorid_last_update
	self.assert_(fetch_bibauthorid_last_update())

	def test_fetch_index_update(self):
	from invenio.bibrank_selfcites_task import fetch_index_update
	self.assert_(fetch_index_update())

	def test_fetch_records(self):
	from invenio.bibrank_selfcites_task import fetch_records
	old_date = datetime(year=1900, month=1, day=1)
	future_date = datetime.now() + timedelta(days=1)
	self.assert_(fetch_records(old_date, future_date))
	self.assert_(not fetch_records(future_date, future_date))

	def test_fetch_concerned_records(self):
	from invenio.bibrank_selfcites_task import fetch_concerned_records, \
	store_last_updated, \
	get_bibrankmethod_lastupdate
	name = 'selfcites'
	old_date = datetime(year=1900, month=1, day=1).strftime("%Y-%m-%d %H:%M:%S")
	try:
	original_date = get_bibrankmethod_lastupdate(name)
	except IndexError:
	original_date = old_date
	store_last_updated(name, old_date)
	self.assert_(fetch_concerned_records('selfcites'))
	future_date = datetime.now() + timedelta(days=1)
	store_last_updated(name, future_date)
	self.assert_(not fetch_concerned_records('selfcites'))
	# Restore value in db
	store_last_updated(name, original_date)

	def test_process_updates(self):
	from invenio.bibrank_selfcites_task import process_updates
	process_updates('selfcites')

	def test_has_algorithms(self):
	from invenio.bibrank_selfcites_indexer import ALL_ALGORITHMS
	self.assert_(ALL_ALGORITHMS)

	def test_process_one(self):
	from invenio.bibrank_selfcites_indexer import get_authors_tags
	from invenio.bibrank_selfcites_task import process_one
	from invenio.bibrank_selfcites_task import get_citations_fun
	from invenio.bibrank_selfcites_indexer import ALL_ALGORITHMS

	tags = get_authors_tags()
	for algorithm in ALL_ALGORITHMS:
	citation_fun = get_citations_fun(algorithm=algorithm)
	process_one(1, tags, citation_fun)

	def test_empty_self_cites_tables(self):
	from invenio.bibrank_selfcites_task import empty_self_cites_tables
	from invenio.dbquery import run_sql
	empty_self_cites_tables()
	counts = [
	run_sql('SELECT count(*) from rnkRECORDSCACHE')[0][0],
	run_sql('SELECT count(*) from rnkEXTENDEDAUTHORS')[0][0],
	run_sql('SELECT count(*) from rnkSELFCITES')[0][0],
	]
	self.assertEqual(counts, [0, 0, 0])

	def test_fill_self_cites_tables(self):
	from invenio.bibrank_selfcites_task import fill_self_cites_tables
	from invenio.dbquery import run_sql
	config = {'algorithm':'friends', 'friends_threshold': 3}
	fill_self_cites_tables(config)
	counts = [
	run_sql('SELECT count(*) from rnkRECORDSCACHE')[0][0],
	run_sql('SELECT count(*) from rnkEXTENDEDAUTHORS')[0][0],
	run_sql('SELECT count(*) from rnkSELFCITES')[0][0],
	]
	self.assert_(counts[0] > 0)
	self.assert_(counts[1] > 0)
	self.assert_(counts[2] > 0)


	TEST_SUITE = make_test_suite(SelfCitesIndexerTests,
	SelfCitesTaskTests)

	if __name__ == "__main__":
	run_test_suite(TEST_SUITE, warn_user=True)

bibrank_selfcites_regression_tests.pyNo OneTemporaryActions

File Metadata

bibrank_selfcites_regression_tests.pyView Options

Event Timeline

bibrank_selfcites_regression_tests.py
No OneTemporary
Actions

bibrank_selfcites_regression_tests.py
View Options