Add a test

DuBibtex.py

@@ -81,7 +81,7 @@ class Parser:
   # current bibitem and bib ID
   cur, bib = None, ''
 
-  def __init__(self):
+  def __init__(self, output_file=None):
     config = configparser.ConfigParser()
     config.read("config.ini")
     Paras.header['User-Agent'] = config.get(Paras.section, "header").strip()
@@ -95,7 +95,7 @@ def __init__(self):
     Paras.inputFileList = config.get(Paras.section,
                                      "inputFileList").strip().split(",")
     Paras.doiJsonFile = config.get(Paras.section, "doiJsonFile").strip()
-    Paras.outputFile = config.get(Paras.section, "outputFile").strip()
+    Paras.outputFile = output_file if output_file else config.get(Paras.section, "outputFile").strip()
     Paras.fieldRemovalList = config.get(Paras.section,
                                         "fieldRemovalList").strip().split(",")
     Paras.minYear = config.getint(Paras.section, "minYear")

pyproject.toml

@@ -0,0 +1,4 @@
+[tool.pytest.ini_options]
+pythonpath = [
+  "."
+]

tests/inputs/.gitignore

@@ -0,0 +1 @@
+!*

tests/inputs/signet.bib

@@ -0,0 +1,8 @@
+@INPROCEEDINGS{feng2021signet,
+  author={Feng, Brandon Yushan and Varshney, Amitabh},
+  booktitle={2021 IEEE/CVF International Conference on Computer Vision (ICCV)}, 
+  title={{SIGNET}: Efficient Neural Representation for Light Fields}, 
+  year={2021},
+  volume={},
+  number={},
+  pages={14204-14213}}

tests/test_find_dois.py

@@ -0,0 +1,33 @@
+import unittest
+import tempfile
+import os
+
+import bibtexparser
+
+from DuBibtex import Parser
+
+
+
+class TestFindDOIs(unittest.TestCase):
+
+
+  def test_iccv_doi(self):
+    filename = "tests/inputs/signet.bib"
+    correct_doi = "10.1109/ICCV48922.2021.01396"
+
+    self.assertTrue(os.path.isfile(filename))
+    library = bibtexparser.parse_file(filename)
+
+    with tempfile.NamedTemporaryFile() as fp:
+        p = Parser(output_file=fp.name)
+        for entry in library.entries:
+          p.copy_from_parsed_entry(entry)
+          p.write_current_item()
+        p.shut_down()
+
+        # Check the doi
+        generated_library = bibtexparser.parse_file(fp.name)
+        self.assertEqual(len(generated_library.entries), len(library.entries), "Number of entries should be the same")
+        for entry in generated_library.entries:
+          self.assertTrue("doi" in entry)
+          self.assertEqual(entry.fields_dict["doi"].value, correct_doi)