add deep compare

tests/test_copy.py

@@ -8,11 +8,48 @@
 import unittest
 
 import aaf2
+from aaf2 import properties
 import common
 
 
+
+
 class AAFCopyTests(unittest.TestCase):
 
+    def compare_objects(self, a, b):
+        for key in a.keys():
+            if a[key].format_name() == "Property":
+                self.assertEqual(a[key].value, b[key].value)
+            else:
+                pa = a[key]
+                pb = b[key]
+
+                if isinstance(pa, (properties.StrongRefSetProperty,
+                                   properties.WeakRefArrayProperty)):
+                    for key, item_a in pa.items():
+                        self.assertTrue(key in pb)
+                        item_b = pb[key]
+                        self.compare_objects(item_a, item_b)
+
+                elif isinstance(pa, properties.StrongRefVectorProperty):
+                    for item_a, item_b in zip(pa, pb):
+                        self.compare_objects(item_a, item_b)
+                elif isinstance(pa, (properties.StrongRefProperty,
+                                     properties.WeakRefProperty)):
+                    self.compare_objects(pa.value, pb.value)
+
+                elif isinstance(pa, properties.StreamProperty):
+                    read_size = a.root.cfb.sector_size
+                    sa = pa.open('r')
+                    sb = pb.open('r')
+                    byte_size = sa.dir.byte_size
+                    self.assertEqual(byte_size, sb.dir.byte_size)
+                    while byte_size > 0:
+                        da = sa.read(read_size)
+                        db = sb.read(read_size)
+                        self.assertEqual(da, db)
+                        byte_size -= len(da)
+
     def test_mob_copy(self):
         test_file = os.path.join(common.test_files_dir(),"test_file_01.aaf")
         result_file = common.get_test_file('copy_mobs.aaf')
@@ -28,6 +65,7 @@ def test_mob_copy(self):
                 for mob in a.content.mobs:
                     other_mob = a.content.mobs.get(mob.mob_id, None)
                     self.assertEqual(mob.mob_id, other_mob.mob_id)
+                    self.compare_objects(mob, other_mob)
 
         result_file = common.get_test_file('copy_mobs_fail.aaf')
         with aaf2.open(test_file, 'r') as a: