Merge pull request sigmavirus24#582 from itsmemattchung/issue/567

sigmavirus24 · sigmavirus24 · commit 584b2cd9d913 · 2016-02-23T14:04:44.000-06:00
Added __eq__ and __ne__ to Tree object
diff --git a/github3/git.py b/github3/git.py
@@ -206,6 +206,12 @@ def _update_attributes(self, tree):
     def _repr(self):
         return '<Tree [{0}]>'.format(self.sha)
 
+    def __eq__(self, other):
+        return self.as_dict() == other.as_dict()
+
+    def __ne__(self, other):
+        return self.as_dict() != other.as_dict()
+
     def recurse(self):
         """Recurse into the tree.
 
diff --git a/tests/cassettes/Tree_ne.json b/tests/cassettes/Tree_ne.json
diff --git a/tests/integration/test_git.py b/tests/integration/test_git.py
@@ -7,6 +7,17 @@ class TestTree(IntegrationHelper):
 
     """Integration tests for methods on the Test class."""
 
+    def test_inequality(self):
+        """Test that a tree and its recursed tree are not equal."""
+        cassette_name = self.cassette_name('ne')
+        with self.recorder.use_cassette(cassette_name):
+            repository = self.gh.repository('sigmavirus24', 'github3.py')
+            tree = repository.tree(
+                '96726db07528a87b7c1f266ed42cd321070470c2'
+            )
+            recursed = tree.recurse()
+            assert tree != recursed
+
     def test_recurse(self):
         """Test recurse on tree"""
         cassette_name = self.cassette_name('recurse')
diff --git a/tests/unit/test_git.py b/tests/unit/test_git.py
@@ -20,6 +20,17 @@ class TestTree(UnitHelper):
     described_class = github3.git.Tree
     example_data = get_example_data()
 
+    def test_eq(self):
+        """Assert that two trees are equal."""
+        tree = github3.git.Tree(get_example_data())
+        assert self.instance == tree
+
+    def test_ne(self):
+        """Assert that two trees are not equal."""
+        tree = github3.git.Tree(get_example_data())
+        tree._json_data['truncated'] = True
+        assert self.instance != tree
+
     def test_repr(self):
         """Assert Tree in in the repr."""
         assert isinstance(self.instance, github3.git.Tree)
@@ -36,7 +47,7 @@ def test_recurse(self):
 
 class TestCommit(UnitHelper):
 
-    "Commit unit test."""
+    """Commit unit test."""
 
     described_class = github3.git.Commit
     example_data = get_commit_example_data()
