adding tests for RecordDawg edges_data() and edgesiter_data()

Author: EliFinkelshteyn

dawg_python/dawgs.py

@@ -537,12 +537,19 @@ def _value_for_index(self, index):
 
     def items(self, prefix=""):
         res = super(RecordDAWG, self).items(prefix)
+        print("items data:")
+        print(res)
         return [(key, self._struct.unpack(val)) for (key, val) in res]
 
     def iteritems(self, prefix=""):
         res = super(RecordDAWG, self).iteritems(prefix)
         return ((key, self._struct.unpack(val)) for (key, val) in res)
 
+    #def edges_data(self, prefix=""):
+    #    return super(RecordDAWG, self).edges_data(prefix)
+
+    #def iteredges_data(self, prefix=""):
+    #    return super(RecordDAWG, self).iteredges_data(prefix)
 
 LOOKUP_ERROR = -1
 

tests/test_payload_dawg.py

@@ -149,6 +149,19 @@ def test_record_items(self):
         d = self.dawg()
         assert d.items() == sorted(self.STRUCTURED_DATA)
 
+    def test_edges_data(self):
+        d = self.dawg()
+        assert d.edges_data('foob') == [('fooba', None)]
+        assert d.edges_data('fooba') == [('foobar', (6, 3, 0))]
+        assert d.edges_data('fo') == [('foo', (3, 2, 1)), ('foo', (3, 2, 256))]
+
+    def test_iteredges_data(self):
+        d = self.dawg()
+        assert list(d.iteredges_data('foob')) == [('fooba', None)]
+        assert list(d.iteredges_data('fooba')) == [('foobar', (6, 3, 0))]
+        assert list(d.iteredges_data('fo')) == [('foo', (3, 2, 1)),
+                                                ('foo', (3, 2, 256))]
+
     def test_record_keys(self):
         d = self.dawg()
         assert d.keys() == ['bar', 'foo', 'foo', 'foobar',]