def test_pmid_to_url(self): # Test data entry_first = { 'title': 'First', 'id': 'aaa'} entry_second = { 'title': 'Second', 'id': 'pmid24685317'} entry_third = { 'title': 'Third', 'id': 'pmid24685317', 'URL': 'http://www.google.com/'} entries = [entry_first, entry_second, entry_third] result = utils.pmid_to_url(entries) self.assertFalse('URL' in result[0]) self.assertEqual(result[1]['URL'], 'http://www.ncbi.nlm.nih.gov/pubmed/24685317') self.assertEqual(result[2]['URL'], entry_third['URL'])
def main(bibtex_file, html_file): """Load BibTeX file and export to WordPress HTML file""" entries = load_bibtex_file(bibtex_file) entries = pmid_to_url(entries) entries = sort_by_date(entries, reverse=True) export_wordpress(html_file, entries)