Add equality testing to Keys too.

sigmavirus24 · sigmavirus24 · commit 8232935b317d · 2013-03-28T18:21:58.000-04:00
diff --git a/github3/gists/gist.py b/github3/gists/gist.py
@@ -21,7 +21,6 @@ class Gist(GitHubCore):
     you own it).  You can also "star" or "unstar" the gist (again assuming you
     have authenticated).
     """
-
     def __init__(self, data, session=None):
         super(Gist, self).__init__(data, session)
         #: Number of comments on this gist
diff --git a/github3/users.py b/github3/users.py
@@ -31,6 +31,9 @@ def __repr__(self):
     def __str__(self):
         return self.key
 
+    def __eq__(self, other):
+        return self.id == other.id
+
     def _update_(self, key):
         self.__init__(key, self._session)
 
diff --git a/tests/test_users.py b/tests/test_users.py
@@ -14,6 +14,9 @@ def setUp(self):
         super(TestKey, self).setUp()
         self.key = github3.users.Key(self.key.to_json(), self.g)
 
+    def test_equality(self):
+        expect(self.key) == github3.users.Key(self.key.to_json())
+
     def test_str(self):
         expect(str(self.key)) == self.key.key
         expect(repr(self.key).startswith('<User Key')).is_True()
