Skip to content

Commit 080202b

Browse files
committed
added custom assertDictEqual() function
1 parent 495c842 commit 080202b

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

test/__init__.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import unittest
66
import json
77
import sys
8+
import collections
89
try:
910
from StringIO import StringIO
1011
except ImportError: # Python 3
@@ -19,6 +20,18 @@
1920

2021
class TestSendGrid(unittest.TestCase):
2122
maxDiff = None
23+
24+
def assertDictEqual(self, d1, d2, msg=None): # assertEqual uses for dicts
25+
for k,v1 in d1.iteritems():
26+
self.assertIn(k, d2, msg)
27+
v2 = d2[k]
28+
if(isinstance(v1, collections.Iterable) and
29+
not isinstance(v1, basestring)):
30+
self.assertItemsEqual(v1, v2, msg)
31+
else:
32+
self.assertEqual(v1, v2, msg)
33+
return True
34+
2235
def setUp(self):
2336
self.sg = SendGridClient(SG_USER, SG_PWD)
2437

0 commit comments

Comments
 (0)