[test/document int] Add section test

Author: mpsonntag

test/test_doc_integration.py

@@ -49,6 +49,10 @@ def save_load(self):
         return jdoc, xdoc, ydoc
 
     def test_id(self):
+        """
+        This test checks the correct writing and loading of
+        autogenerated and assigned document id.
+        """
         # Test correct save and load of generated id.
         jdoc, xdoc, ydoc = self.save_load()
 
@@ -66,6 +70,10 @@ def test_id(self):
         self.assertEqual(ydoc.id, assigned_id)
 
     def test_simple_attributes(self):
+        """
+        This test checks correct writing and loading of 'simple'
+        Document format attributes.
+        """
         author = "HPL"
         version = "ver64"
         date = "1890-08-20"
@@ -91,3 +99,68 @@ def test_simple_attributes(self):
         self.assertEqual(ydoc.version, version)
         self.assertEqual(str(ydoc.date), date)
         self.assertEqual(ydoc.repository, repository)
+
+    def test_children(self):
+        """
+        This test checks the correct saving and loading of Section children of a Document.
+        """
+        # Lvl 1 child Sections
+        sec_lvl_11 = odml.Section(name="sec_11", parent=self.doc)
+        _ = odml.Section(name="sec_12", parent=self.doc)
+
+        # Lvl 2 child Sections
+        sec_lvl_21 = odml.Section(name="sec_21", parent=sec_lvl_11)
+        _ = odml.Section(name="sec_22", parent=sec_lvl_11)
+        _ = odml.Section(name="sec_23", parent=sec_lvl_11)
+
+        # Lvl 2 child Properties
+        _ = odml.Property(name="prop_21", parent=sec_lvl_11)
+        _ = odml.Property(name="prop_22", parent=sec_lvl_11)
+        _ = odml.Property(name="prop_23", parent=sec_lvl_11)
+
+        # Lvl 3 child Sections
+        _ = odml.Section(name="sec_31", parent=sec_lvl_21)
+        _ = odml.Section(name="sec_32", parent=sec_lvl_21)
+        _ = odml.Section(name="sec_33", parent=sec_lvl_21)
+        _ = odml.Section(name="sec_34", parent=sec_lvl_21)
+
+        # Lvl 3 child Properties
+        _ = odml.Property(name="prop_31", parent=sec_lvl_21)
+        _ = odml.Property(name="prop_32", parent=sec_lvl_21)
+        _ = odml.Property(name="prop_33", parent=sec_lvl_21)
+        _ = odml.Property(name="prop_34", parent=sec_lvl_21)
+
+        jdoc, xdoc, ydoc = self.save_load()
+
+        # Test correct JSON save and load.
+        self.assertEqual(len(jdoc.sections), 2)
+
+        jsec_lvl_1 = jdoc[sec_lvl_11.name]
+        self.assertEqual(len(jsec_lvl_1.sections), 3)
+        self.assertEqual(len(jsec_lvl_1.properties), 3)
+
+        jsec_lvl_2 = jsec_lvl_1[sec_lvl_21.name]
+        self.assertEqual(len(jsec_lvl_2.sections), 4)
+        self.assertEqual(len(jsec_lvl_2.properties), 4)
+
+        # Test correct XML save and load.
+        self.assertEqual(len(xdoc.sections), 2)
+
+        xsec_lvl_1 = xdoc[sec_lvl_11.name]
+        self.assertEqual(len(xsec_lvl_1.sections), 3)
+        self.assertEqual(len(xsec_lvl_1.properties), 3)
+
+        xsec_lvl_2 = xsec_lvl_1[sec_lvl_21.name]
+        self.assertEqual(len(xsec_lvl_2.sections), 4)
+        self.assertEqual(len(xsec_lvl_2.properties), 4)
+
+        # Test correct YAML save and load.
+        self.assertEqual(len(ydoc.sections), 2)
+
+        ysec_lvl_1 = ydoc[sec_lvl_11.name]
+        self.assertEqual(len(ysec_lvl_1.sections), 3)
+        self.assertEqual(len(ysec_lvl_1.properties), 3)
+
+        ysec_lvl_2 = ysec_lvl_1[sec_lvl_21.name]
+        self.assertEqual(len(ysec_lvl_2.sections), 4)
+        self.assertEqual(len(ysec_lvl_2.properties), 4)