[test/versionConv int] Add yaml test

mpsonntag · mpsonntag · commit 9ea6feca2b96 · 2018-04-23T22:14:29.000+02:00
diff --git a/test/resources/version_conversion_int.yaml b/test/resources/version_conversion_int.yaml
@@ -0,0 +1,195 @@
+Document:
+  author: author
+  date: '2018-02-02'
+  repository: http://portal.g-node.org/odml/terminologies/v1.0/terminologies.xml
+  sections:
+  - definition: def s1
+    include: url s1
+    link: somewhere s1
+    name: sec_one
+    properties:
+    - definition: def prop1
+      dependency: dep p1
+      dependency_value: dep val p1
+      name: prop_one
+      values:
+      - dtype: string
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - string
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 11
+        unit: arbitrary
+        value: '1'
+      - definition: def val 2
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 2
+        reference: ref val 2
+        uncertainty: 12
+        unit: arbitrary 2
+        value: '2'
+      - definition: def val 3
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - string
+        filename: filename val 3
+        reference: ref val 3
+        uncertainty: 13
+        unit: arbitrary 3
+        value: '3'
+    - definition: def prop2
+      dependency: dep p2
+      dependency_value: dep val p2
+      name: prop_two
+      values:
+      - dtype: string
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '1'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '2'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '3'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '4'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '4'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '4'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '4'
+      - definition: def val 1
+        dtype: !!python/object/apply:odml.dtypes.DType
+        - int
+        filename: filename val 1
+        reference: ref val 1
+        uncertainty: 12
+        unit: arbitrary
+        value: '4'
+    - definition: def prop3
+      dependency: dep p3
+      dependency_value: dep val p3
+      name: prop_three
+      values:
+      - dtype: string
+    reference: ref s1
+    sections:
+    - definition: def subs1
+      include: url subs1
+      link: somewhere subs1
+      name: subsection one
+      properties:
+      - definition: def prop1
+        dependency: dep p1
+        dependency_value: dep val p1
+        name: prop one
+        values:
+        - dtype: string
+      - definition: def prop3
+        dependency: dep p3
+        dependency_value: dep val p3
+        name: prop three
+        values:
+        - dtype: string
+        - definition: def val 1
+          dtype: !!python/object/apply:odml.dtypes.DType
+          - int
+          filename: filename val 1
+          reference: ref val 1
+          uncertainty: 12
+          unit: arbitrary
+          value: '4'
+        - definition: def val 1
+          dtype: !!python/object/apply:odml.dtypes.DType
+          - int
+          filename: filename val 1
+          reference: ref val 1
+          uncertainty: 12
+          unit: arbitrary
+          value: '4'
+        - definition: def val 1
+          dtype: !!python/object/apply:odml.dtypes.DType
+          - int
+          filename: filename val 1
+          reference: ref val 1
+          uncertainty: 12
+          unit: arbitrary
+          value: '4'
+        - definition: def val 1
+          dtype: !!python/object/apply:odml.dtypes.DType
+          - int
+          filename: filename val 1
+          reference: ref val 1
+          uncertainty: 12
+          unit: arbitrary
+          value: '2'
+      - definition: def prop2
+        dependency: dep p2
+        dependency_value: dep val p2
+        name: prop two
+        values:
+        - dtype: string
+      reference: ref subs1
+      sections: []
+      type: subsec
+    type: mainsec
+  - definition: def s2
+    include: url s2
+    link: somewhere s2
+    name: section two
+    properties: []
+    reference: ref s2
+    sections: []
+    type: mainsec
+  - definition: def s3
+    include: url s3
+    link: somewhere s3
+    name: section three
+    properties: []
+    reference: ref s3
+    sections: []
+    type: mainsec
+  version: v1.13
+odml-version: '1'
diff --git a/test/test_version_converter_integration.py b/test/test_version_converter_integration.py
@@ -16,21 +16,22 @@ class TestVersionConverterIntegration(unittest.TestCase):
     def setUp(self):
         # Set up test environment
         dir_path = os.path.dirname(os.path.realpath(__file__))
+
         self.xmlfile = os.path.join(dir_path, "resources", "version_conversion_int.xml")
+        self.yamlfile = os.path.join(dir_path, "resources", "version_conversion_int.yaml")
 
         self.tmp_dir = tempfile.mkdtemp(suffix=".odml")
+        self.outfile = os.path.join(self.tmp_dir, "version_conversion.xml")
 
     def tearDown(self):
         if os.path.exists(self.tmp_dir):
             shutil.rmtree(self.tmp_dir)
 
     def test_convert_xml(self):
-        outfile = os.path.join(self.tmp_dir, "version_conversion.xml")
-
-        VC(self.xmlfile).write_to_file(outfile)
-        self.assertTrue(os.path.exists(outfile))
+        VC(self.xmlfile).write_to_file(self.outfile)
+        self.assertTrue(os.path.exists(self.outfile))
 
-        doc = load(outfile)
+        doc = load(self.outfile)
 
         repo = "http://portal.g-node.org/odml/terminologies/v1.1/terminologies.xml"
         self.assertEqual(doc.repository, repo)
@@ -61,3 +62,43 @@ def test_convert_xml(self):
 
         prop = sec.properties["prop_three"]
         self.assertEqual(len(prop.value), 0)
+
+    def test_convert_yaml(self):
+        with self.assertRaises(Exception):
+            VC(self.yamlfile).write_to_file(self.outfile)
+
+        VC(self.yamlfile).write_to_file(self.outfile, "YAML")
+        self.assertTrue(os.path.exists(self.outfile))
+
+        doc = load(self.outfile)
+
+        # Test document attribute export
+        self.assertEqual(doc.author, "author")
+        self.assertEqual(doc.version, "v1.13")
+        repo = "http://portal.g-node.org/odml/terminologies/v1.1/terminologies.xml"
+        self.assertEqual(doc.repository, repo)
+        self.assertEqual(len(doc.sections), 3)
+
+        sec = doc.sections["sec_one"]
+        self.assertEqual(sec.definition, "def s1")
+        self.assertEqual(sec.reference, "ref s1")
+        self.assertEqual(sec.type, "mainsec")
+        self.assertEqual(len(sec.sections), 1)
+        self.assertEqual(len(sec.properties), 3)
+
+        prop = sec.properties["prop_one"]
+        self.assertEqual(prop.definition, "def prop1")
+        self.assertEqual(prop.dependency, "dep p1")
+        self.assertEqual(prop.dependency_value, "dep val p1")
+        self.assertEqual(prop.reference, "ref val 1")
+        self.assertEqual(prop.uncertainty, "11")
+        self.assertEqual(prop.unit, "arbitrary")
+        self.assertEqual(prop.value_origin, "filename val 1")
+        self.assertEqual(prop.dtype, "string")
+        self.assertEqual(len(prop.value), 3)
+
+        prop = sec.properties["prop_two"]
+        self.assertEqual(len(prop.value), 8)
+
+        prop = sec.properties["prop_three"]
+        self.assertEqual(len(prop.value), 0)
