Spaces:
Running
Running
fix tests
Browse files
tests/test_grobid_processors.py
CHANGED
@@ -38,7 +38,7 @@ def test_get_xml_nodes_header_paragraphs():
|
|
38 |
|
39 |
children = get_xml_nodes_header(soup)
|
40 |
|
41 |
-
assert len(children) == 8
|
42 |
|
43 |
|
44 |
def test_get_xml_nodes_header_sentences():
|
@@ -47,4 +47,4 @@ def test_get_xml_nodes_header_sentences():
|
|
47 |
|
48 |
children = get_xml_nodes_header(soup, use_paragraphs=False)
|
49 |
|
50 |
-
assert len(children) == 15
|
|
|
38 |
|
39 |
children = get_xml_nodes_header(soup)
|
40 |
|
41 |
+
assert sum([len(child) for k, child in children.items()]) == 8
|
42 |
|
43 |
|
44 |
def test_get_xml_nodes_header_sentences():
|
|
|
47 |
|
48 |
children = get_xml_nodes_header(soup, use_paragraphs=False)
|
49 |
|
50 |
+
assert sum([len(child) for k, child in children.items()]) == 15
|