diff --git a/README.md b/README.md index 55574dd..5ace1c5 100644 --- a/README.md +++ b/README.md @@ -47,3 +47,4 @@ Käynnistä ohjelma: python src/index.py ``` + diff --git a/src/tests/bibtex_generation_test.py b/src/tests/bibtex_generation_test.py index 144a573..14e5921 100644 --- a/src/tests/bibtex_generation_test.py +++ b/src/tests/bibtex_generation_test.py @@ -2,26 +2,27 @@ from entities.citation import Citation from util import generate_bibtex + class TestBibtexGeneration(unittest.TestCase): def setUp(self): pass def test_bibtex_generates_correctly(self): citations = [] - citations.append(Citation("article", "1", - { - "author": "1", - "title": "1", - "year": "1", - "journal": "1" - })) - citations.append(Citation("article", "2", - { - "author": "2", - "title": "2", - "year": "2", - "journal": "2" - })) + citations.append( + Citation( + "article", + "1", + {"author": "1", "title": "1", "year": "1", "journal": "1"}, + ) + ) + citations.append( + Citation( + "article", + "2", + {"author": "2", "title": "2", "year": "2", "journal": "2"}, + ) + ) bibtex = generate_bibtex(citations) @@ -29,14 +30,14 @@ def test_bibtex_generates_correctly(self): \tauthor = {1}, \ttitle = {1}, \tyear = {1}, -\tjournal = {1}, +\tjournal = {1} } @article{2, \tauthor = {2}, \ttitle = {2}, \tyear = {2}, -\tjournal = {2}, +\tjournal = {2} } """ diff --git a/src/util.py b/src/util.py index 2e57d9d..1252169 100644 --- a/src/util.py +++ b/src/util.py @@ -40,7 +40,8 @@ def generate_bibtex(citations: list[Citation]): bibtex += f"@{citation.citation_type}{{{citation.citation_key},\n" for field_name, field_value in citation.fields.items(): bibtex += f"\t{field_name} = {{{field_value}}},\n" - bibtex += "}\n\n" + bibtex = bibtex[:-2] + bibtex += "\n}\n\n" return bibtex