Convert SVG metadata test to XML/XPath parsing.

QuLogic · QuLogic · commit 207013d1b2a6 · 2020-06-09T23:22:16.000-04:00
diff --git a/lib/matplotlib/tests/test_backend_svg.py b/lib/matplotlib/tests/test_backend_svg.py
@@ -1,8 +1,8 @@
 import datetime
 from io import BytesIO
 import re
-import sys
 import tempfile
+import xml.etree.ElementTree
 import xml.parsers.expat
 
 import numpy as np
@@ -294,16 +294,41 @@ def test_svg_metadata():
         fig.savefig(fd, format='svg', metadata=metadata)
         buf = fd.getvalue().decode()
 
-    # Check things that are easy, single or multi-value entries.
+    SVGNS = '{http://www.w3.org/2000/svg}'
+    RDFNS = '{http://www.w3.org/1999/02/22-rdf-syntax-ns#}'
+    CCNS = '{http://creativecommons.org/ns#}'
+    DCNS = '{http://purl.org/dc/elements/1.1/}'
+
+    root = xml.etree.ElementTree.fromstring(buf)
+    rdf, = root.findall(f'./{SVGNS}metadata/{RDFNS}RDF')
+
+    # Check things that are single entries.
+    titles = [node.text for node in root.findall(f'./{SVGNS}title')]
+    assert titles == [metadata['Title']]
+    types = [node.attrib[f'{RDFNS}resource']
+             for node in rdf.findall(f'./{CCNS}Work/{DCNS}type')]
+    assert types == [metadata['Type']]
     for k in ['Description', *single_value]:
-        if sys.platform == 'win32':
-            assert metadata[k].replace('\n', '\r\n') in buf
-        else:
-            assert metadata[k] in buf
+        if k == 'Type':
+            continue
+        values = [node.text
+                  for node in rdf.findall(f'./{CCNS}Work/{DCNS}{k.lower()}')]
+        assert values == [metadata[k]]
+
+    # Check things that are multi-value entries.
     for k in multi_value:
-        for v in metadata[k]:
-            assert v in buf
+        if k == 'Keywords':
+            continue
+        values = [
+            node.text
+            for node in rdf.findall(
+                f'./{CCNS}Work/{DCNS}{k.lower()}/{CCNS}Agent/{DCNS}title')]
+        assert values == metadata[k]
 
     # Check special things.
-    assert '1968-08-01' in buf  # Start date
-    assert '1968-08-02T01:02:03' in buf  # End date
+    dates = [node.text for node in rdf.findall(f'./{CCNS}Work/{DCNS}date')]
+    assert dates == ['1968-08-01/1968-08-02T01:02:03']
+
+    values = [node.text for node in
+              rdf.findall(f'./{CCNS}Work/{DCNS}subject/{RDFNS}Bag/{RDFNS}li')]
+    assert values == metadata['Keywords']
