Merge pull request #11 from ruofeidu/more_tests

More tests

tests/inputs/qdiffusion.bib

@@ -0,0 +1,8 @@
+@InProceedings{li2023qdiffusion,
+  author={Li, Xiuyu and Liu, Yijiang and Lian, Long and Yang, Huanrui and Dong, Zhen and Kang, Daniel and Zhang, Shanghang and Keutzer, Kurt},
+  title={Q-Diffusion: Quantizing Diffusion Models},
+  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
+  month={October},
+  year={2023},
+  pages={17535-17545}
+}

tests/test_find_dois.py

@@ -1,33 +1,33 @@
-import unittest
 import tempfile
 import os
 
 import bibtexparser
+import pytest
 
 from DuBibtex import Parser
 
 
+@pytest.mark.parametrize("filename,correct_doi", [
+    ("signet.bib", "10.1109/ICCV48922.2021.01396"),
+    ("qdiffusion.bib", "10.1109/ICCV51070.2023.01608"),
+])
+def test_iccv_doi(filename, correct_doi):
+    input_file = os.path.join("tests", "inputs", filename)
 
-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)
+    assert os.path.isfile(input_file), f"File {input_file} does not exist"
+    library = bibtexparser.parse_file(input_file)
 
     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.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")
+        assert 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)
+            assert "doi" in entry
+            assert entry.fields_dict["doi"].value == correct_doi