def test_data2group(self): right = [ ["10 mbox.example.com"], [ "10 mbox1.example.com", "20 mbox2.example.com", "20 mbox3.example.com", "30 mbox4.example.com", "30 mbox5.example.com", "30 mbox6.example.com", ], ] rschema = OrderedDict((("prio", int), ("srvr", str),)) results = [ OrderedDict([(10, ["mbox.example.com"])]), OrderedDict( [ (10, ["mbox1.example.com"]), (20, ["mbox2.example.com", "mbox3.example.com"]), ( 30, ["mbox4.example.com", "mbox5.example.com", "mbox6.example.com"], ), ] ), ] for rdata, res in zip(right, results): group = _data2rec_group(rschema, rdata, "prio") self.assertEqual(group, res)
def test_data2group(self): right = [ ['10 mbox.example.com'], [ '10 mbox1.example.com', '20 mbox2.example.com', '20 mbox3.example.com', '30 mbox4.example.com', '30 mbox5.example.com', '30 mbox6.example.com', ], ] rschema = OrderedDict(( ('prio', int), ('srvr', str), )) results = [ OrderedDict([(10, [{'srvr': 'mbox.example.com'}])]), OrderedDict([ (10, [{'srvr': 'mbox1.example.com'}]), (20, [{'srvr': 'mbox2.example.com'}, {'srvr': 'mbox3.example.com'}]), (30, [{'srvr': 'mbox4.example.com'}, {'srvr': 'mbox5.example.com'}, {'srvr': 'mbox6.example.com'}])] ), ] for rdata, res in zip(right, results): group = _data2rec_group(rschema, rdata, 'prio') self.assertEqual(group, res)