to_legacy_dict() helper, and start adding some new fields

1 files changed, 2 insertions, 13 deletions

diff --git a/tests/test_parse.py b/tests/test_parse.py
index 30b2926..825b561 100644
--- a/tests/test_parse.py
+++ b/tests/test_parse.py
@@ -75,25 +75,14 @@ def test_small_xml() -> None:
     assert doc == expected
 
 
-def test_small_xml_json() -> None:
+def test_small_xml_legacy() -> None:
 
     with open('tests/files/small.xml', 'r') as f:
         tei_xml = f.read()
     with open('tests/files/small.json', 'r') as f:
         json_form = json.loads(f.read())
 
-    d = parse_document_xml(tei_xml).to_dict()
-
-    # munge back to the old JSON format
-    d.update(d.pop('header'))
-    addr = d['authors'][0]['affiliation']['address']
-    addr['postCode'] = addr.pop('post_code')
-
-    # remove nulls from old JSON
-    for c in json_form['citations']:
-        for k in list(c.keys()):
-            if c[k] is None:
-                c.pop(k)
+    d = parse_document_xml(tei_xml).to_legacy_dict()
 
     assert d == json_form