lfoppiano commited on
Commit
93404c2
1 Parent(s): 93bc8ec
Files changed (1) hide show
  1. tests/test_grobid_processors.py +2 -2
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