def setUp(self):
self.db_path = "db/tmp_test"
# make sure we start over fresh
if os.path.exists(self.db_path):
os.remove(self.db_path)
self.db = Database(self.db_path)
self.db.load()
class TestSearchMethods(unittest.TestCase):
def setUp(self):
self.db_path = "db/tmp_test"
# make sure we start over fresh
if os.path.exists(self.db_path):
os.remove(self.db_path)
self.db = Database(self.db_path)
self.db.load()
def test_qearch_query(self):
self.assertEqual(self.db.get_recids_from_query("recid:1"), {'1'})
# make sure those aren't likely to change with time
search_query_result_length = {
"a feynman and date < 1990": 76,
"a gell-mann and date < 1990": 123
}
for query, length in search_query_result_length.items():
with self.subTest(query=query):
recids = self.db.get_recids_from_query(query)
# GreaterEqual because we can get more citations with time...
self.assertGreaterEqual(len(recids), length)
def test_bibkey_search(self):
recid = "1"
record = self.db.get_record(recid)
record.bibkey = "asdf:2010x"
self.db.update_record(recid, record)
results = self.db.get_recids_from_bibkeys({record.bibkey, "notinhere"},
offline_only=True)
self.assertEqual(results, {record.bibkey: recid})
results = self.db.get_recids_from_bibkeys({"Davies:2016ruz"})
self.assertEqual(results, {"Davies:2016ruz": "1472971"})
def test_save_load(self):
# add phony record
record = self.db.get_record(self.test_recid)
record.citations_dl = True
record.citations = {"1", "2", "3"}
record.bibkey = "asdf:2010x"
record.custom_label = "blargh"
self.db.update_record(self.test_recid, record)
self.db.save()
self.assertTrue(os.path.exists(self.db_path))
db2 = Database(self.db.backup_path)
db2.load()
self.assertEqual(len(db2._records), len(self.db._records))
for recid, record in db2._records.items():
self.assertEqual(record, self.db.get_record(recid))
class TestDatabase(unittest.TestCase):
def setUp(self):
# note: sometimes this is not exactly the same on the /references
# page, as we only get those for papers which are registered
# on inspirehep.
# we test with ">=", since this can still change
self.expected_recid_references = {
"0699123": 21,
"460528": 0,
"1514038": 119,
"44993": 0
}
self.expected_recid_citations = {
"0699123": 104,
"460528": 5,
"1514038": 2,
"44993": 0
}
self.db_path = "db/tmp_test"
# make sure we start over fresh
if os.path.exists(self.db_path):
os.remove(self.db_path)
self.db = Database(self.db_path)
self.db.load()
def test_get_references(self):
for recid, expected in self.expected_recid_references.items():
with self.subTest(recid=recid):
self._test_get_references(recid, expected)
def test_get_citations(self):
for recid, expected in self.expected_recid_citations.items():
with self.subTest(recid=recid):
self._test_get_citations(recid, expected)
def _test_get_references(self, recid, expected_references):
record = self.db.get_record(recid)
self.assertFalse(record.references_dl)
self.assertEqual(len(record.references), 0)
self.db.get_references(recid)
self.assertGreaterEqual(len(record.references), expected_references)
self.assertTrue(record.references_dl)
def _test_get_citations(self, recid, expected_citations):
record = self.db.get_record(recid)
self.assertFalse(record.citations_dl)
self.db.get_citations(recid)
self.assertGreaterEqual(len(record.citations), expected_citations)
self.assertTrue(record.citations_dl)
def setUp(self):
# note: sometimes this is not exactly the same on the /references
# page, as we only get those for papers which are registered
# on inspirehep.
# we test with ">=", since this can still change
self.expected_recid_references = {
"0699123": 21,
"460528": 0,
"1514038": 119,
"44993": 0
}
self.expected_recid_citations = {
"0699123": 104,
"460528": 5,
"1514038": 2,
"44993": 0
}
self.db_path = "db/tmp_test"
# make sure we start over fresh
if os.path.exists(self.db_path):
os.remove(self.db_path)
self.db = Database(self.db_path)
self.db.load()
class TestDatabaseBasics(unittest.TestCase):
# things that don't need to download anything
def setUp(self):
self.db_path = "db/tmp_test"
self.test_recid = "123456"
# make sure we start over fresh
if os.path.exists(self.db_path):
os.remove(self.db_path)
self.db = Database(self.db_path)
self.db.load()
def test_empty(self):
self.assertEqual(len(self.db._records), 0)
self.db.load()
self.db.get_record(self.test_recid)
self.assertEqual(len(self.db._records), 1)
r = self.db._records[self.test_recid]
self.assertEqual(r.recid, self.test_recid)
def test_save_load(self):
# add phony record
record = self.db.get_record(self.test_recid)
record.citations_dl = True
record.citations = {"1", "2", "3"}
record.bibkey = "asdf:2010x"
record.custom_label = "blargh"
self.db.update_record(self.test_recid, record)
self.db.save()
self.assertTrue(os.path.exists(self.db_path))
db2 = Database(self.db.backup_path)
db2.load()
self.assertEqual(len(db2._records), len(self.db._records))
for recid, record in db2._records.items():
self.assertEqual(record, self.db.get_record(recid))
logcontrol.set_verbosity_from_argparse(args.verbosity)
# fixme: add tests again
if args.plot and not args.output:
logger.critical("We need output filename to plot. Exiting.")
sys.exit(20)
# todo: maybe use a proper format to save the record data or at least allow ...
# .... to export into such
# todo: add clusters
# todo: extract more infomration; add title as tooltip
# fixme: Restore .travis to specific tests again.
db = Database(args.database[0])
db.load(args.database)
db.statistics()
# get recids
recids = set()
recids.update(get_recid_from_queries(args.queries, db=db))
recids.update(get_recids_from_bibkey_paths(args.bibkeypaths, db=db))
recids.update(get_recids_from_recid_paths(args.recidpaths))
recids.update(get_recids_from_url_paths(args.urlpaths))
db.autocomplete_records(args.get, force=args.forceupdate, recids=recids)
if args.labels:
db.get_labels_from_file(args.labels)