make fmt (black 21.9b0)

2 files changed, 52 insertions, 42 deletions

diff --git a/tests/test_csl.py b/tests/test_csl.py
index 27c8c3e..e8ded91 100644
--- a/tests/test_csl.py
+++ b/tests/test_csl.py
@@ -3,7 +3,7 @@ from grobid_tei_xml import parse_document_xml
 
 def test_small_xml_csl() -> None:
 
-    with open('tests/files/small.xml', 'r') as f:
+    with open("tests/files/small.xml", "r") as f:
         tei_xml = f.read()
 
     d = parse_document_xml(tei_xml)
@@ -11,10 +11,7 @@ def test_small_xml_csl() -> None:
         "type": "article-journal",
         "title": "Dummy Example File",
         "author": [
-            {
-                "given": "Brewster",
-                "family": "Kahle"
-            },
+            {"given": "Brewster", "family": "Kahle"},
             {
                 "given": "J",
                 "family": "Doe",
@@ -29,10 +26,7 @@ def test_small_xml_csl() -> None:
         "type": "article-journal",
         "title": "Everything is Wonderful",
         "author": [
-            {
-                "given": "A",
-                "family": "Seaperson"
-            },
+            {"given": "A", "family": "Seaperson"},
         ],
         "container-title": "Letters in the Alphabet",
         "issued": [[2001]],
diff --git a/tests/test_parse.py b/tests/test_parse.py
index 976d1b1..25529c4 100644
--- a/tests/test_parse.py
+++ b/tests/test_parse.py
@@ -11,31 +11,33 @@ from grobid_tei_xml.types import *
 
 def test_small_xml() -> None:
 
-    with open('tests/files/small.xml', 'r') as f:
+    with open("tests/files/small.xml", "r") as f:
         tei_xml = f.read()
 
     doc = parse_document_xml(tei_xml)
     expected_body = """Introduction \nEverything starts somewhere, as somebody [1]  once said. \n\n In Depth \n Meat \nYou know, for kids. \n Potatos \nQED."""
     expected = GrobidDocument(
-        grobid_version='0.5.1-SNAPSHOT',
-        grobid_timestamp='2018-04-02T00:31+0000',
-        language_code='en',
+        grobid_version="0.5.1-SNAPSHOT",
+        grobid_timestamp="2018-04-02T00:31+0000",
+        language_code="en",
         header=GrobidBiblio(
             title="Dummy Example File",
             authors=[
-                GrobidAuthor(full_name="Brewster Kahle",
-                             given_name="Brewster",
-                             surname="Kahle",
-                             affiliation=GrobidAffiliation(
-                                 department="Faculty ofAgricultrial Engineering",
-                                 laboratory="Plant Physiology Laboratory",
-                                 institution="Technion-Israel Institute of Technology",
-                                 address=GrobidAddress(
-                                     post_code="32000",
-                                     settlement="Haifa",
-                                     country="Israel",
-                                 ),
-                             )),
+                GrobidAuthor(
+                    full_name="Brewster Kahle",
+                    given_name="Brewster",
+                    surname="Kahle",
+                    affiliation=GrobidAffiliation(
+                        department="Faculty ofAgricultrial Engineering",
+                        laboratory="Plant Physiology Laboratory",
+                        institution="Technion-Israel Institute of Technology",
+                        address=GrobidAddress(
+                            post_code="32000",
+                            settlement="Haifa",
+                            country="Israel",
+                        ),
+                    ),
+                ),
                 GrobidAuthor(
                     full_name="J Doe",
                     given_name="J",
@@ -80,9 +82,9 @@ def test_small_xml() -> None:
 
 def test_small_xml_legacy() -> None:
 
-    with open('tests/files/small.xml', 'r') as f:
+    with open("tests/files/small.xml", "r") as f:
         tei_xml = f.read()
-    with open('tests/files/small.json', 'r') as f:
+    with open("tests/files/small.json", "r") as f:
         json_form = json.loads(f.read())
 
     d = parse_document_xml(tei_xml).to_legacy_dict()
@@ -104,7 +106,7 @@ def test_invalid_xml() -> None:
 
 def test_bytes() -> None:
 
-    with open('tests/files/small.xml', 'rb') as f:
+    with open("tests/files/small.xml", "rb") as f:
         tei_xml = f.read()
 
     parse_document_xml(tei_xml)
@@ -113,7 +115,7 @@ def test_bytes() -> None:
 
 def test_elementtree() -> None:
 
-    with open('tests/files/small.xml', 'rb') as f:
+    with open("tests/files/small.xml", "rb") as f:
         tei_xml = f.read()
 
     parse_document_xml(xml.etree.ElementTree.parse(io.BytesIO(tei_xml)))  # type: ignore
@@ -126,8 +128,10 @@ def test_example_grobid_tei_xml() -> None:
 
     doc = parse_document_xml(blob)
 
-    assert doc.header.title == \
-        """Changes of patients' satisfaction with the health care services in Lithuanian Health Promoting Hospitals network"""
+    assert (
+        doc.header.title
+        == """Changes of patients' satisfaction with the health care services in Lithuanian Health Promoting Hospitals network"""
+    )
 
     ref = [c for c in doc.citations or [] if c.id == "b12"][0]
     assert ref.authors[0].full_name == "K Tasa"
@@ -138,8 +142,10 @@ def test_example_grobid_tei_xml() -> None:
     assert ref.date == "1996"
     assert ref.pages == "206-225"
     assert ref.volume == "8"
-    assert ref.unstructured == \
-        """Tasa K, Baker R, Murray M. Using patient feedback for qua- lity improvement. Quality Management in Health Care 1996;8:206-19."""
+    assert (
+        ref.unstructured
+        == """Tasa K, Baker R, Murray M. Using patient feedback for qua- lity improvement. Quality Management in Health Care 1996;8:206-19."""
+    )
 
 
 def test_single_citations_xml() -> None:
@@ -193,7 +199,10 @@ def test_single_citations_xml() -> None:
 </biblStruct>"""
 
     d = parse_citations_xml(citation_xml)[0]
-    assert d.title == """Mesh migration following abdominal hernia repair: a comprehensive review"""
+    assert (
+        d.title
+        == """Mesh migration following abdominal hernia repair: a comprehensive review"""
+    )
     assert d.authors[2].given_name == "L"
     assert d.authors[2].middle_name == "R"
     assert d.authors[2].surname == "Taveras"
@@ -211,7 +220,7 @@ def test_single_citations_xml() -> None:
 
 def test_citation_list_xml() -> None:
 
-    with open('tests/files/example_citation_list.xml', 'r') as f:
+    with open("tests/files/example_citation_list.xml", "r") as f:
         tei_xml = f.read()
 
     citations = parse_citations_xml(tei_xml)
@@ -237,17 +246,21 @@ def test_citation_list_xml() -> None:
     assert citations[11].series_title == "Handbook of Optics"
     assert citations[11].publisher == "McGRAW-HILL"
 
-    assert citations[
-        12].title == "Implications of abandoned shoreline features above Glacial Lake Duluth levels along the north shore of the Superior Basin in the vicinity of the Brule River"
-    assert citations[
-        12].book_title == "Paper presented at the 13th Biennial Meeting of the American Quaternary Association"
+    assert (
+        citations[12].title
+        == "Implications of abandoned shoreline features above Glacial Lake Duluth levels along the north shore of the Superior Basin in the vicinity of the Brule River"
+    )
+    assert (
+        citations[12].book_title
+        == "Paper presented at the 13th Biennial Meeting of the American Quaternary Association"
+    )
     assert citations[12].institution == "University of Minnesota"
 
 
 def test_grobid_070_document() -> None:
     # more recent GROBID v0.7.0 output
 
-    with open('tests/files/example_grobid_plos.tei.xml', 'r') as f:
+    with open("tests/files/example_grobid_plos.tei.xml", "r") as f:
         tei_xml = f.read()
 
     doc = parse_document_xml(tei_xml)
@@ -267,7 +280,10 @@ def test_grobid_070_document() -> None:
 
     cite_b3 = doc.citations[3]
     assert cite_b3.url == "http://unesdoc.unesco.org/ulis/"
-    assert cite_b3.title == "Requirements for Global Implementation of the Strategic Plan for Coastal GOOS"
+    assert (
+        cite_b3.title
+        == "Requirements for Global Implementation of the Strategic Plan for Coastal GOOS"
+    )
     assert cite_b3.authors
     assert cite_b3.authors[0].surname == "Ioc-Unesco"
     assert cite_b3.date == "2012"