Revert moving tests out of 'git' folder, related to #1030

1 files changed, 108 insertions, 0 deletions

diff --git a/git/test/performance/test_commit.py b/git/test/performance/test_commit.py
new file mode 100644
index 00000000..578194a2
--- /dev/null
+++ b/git/test/performance/test_commit.py
@@ -0,0 +1,108 @@
+# test_performance.py
+# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
+#
+# This module is part of GitPython and is released under
+# the BSD License: http://www.opensource.org/licenses/bsd-license.php
+from __future__ import print_function
+from io import BytesIO
+from time import time
+import sys
+
+from .lib import TestBigRepoRW
+from git import Commit
+from gitdb import IStream
+from git.test.test_commit import TestCommitSerialization
+
+
+class TestPerformance(TestBigRepoRW, TestCommitSerialization):
+
+    def tearDown(self):
+        import gc
+        gc.collect()
+
+    # ref with about 100 commits in its history
+    ref_100 = '0.1.6'
+
+    def _query_commit_info(self, c):
+        c.author
+        c.authored_date
+        c.author_tz_offset
+        c.committer
+        c.committed_date
+        c.committer_tz_offset
+        c.message
+        c.parents
+
+    def test_iteration(self):
+        no = 0
+        nc = 0
+
+        # find the first commit containing the given path - always do a full
+        # iteration ( restricted to the path in question ), but in fact it should
+        # return quite a lot of commits, we just take one and hence abort the operation
+
+        st = time()
+        for c in self.rorepo.iter_commits(self.ref_100):
+            nc += 1
+            self._query_commit_info(c)
+            for obj in c.tree.traverse():
+                obj.size
+                no += 1
+            # END for each object
+        # END for each commit
+        elapsed_time = time() - st
+        print("Traversed %i Trees and a total of %i uncached objects in %s [s] ( %f objs/s )"
+              % (nc, no, elapsed_time, no / elapsed_time), file=sys.stderr)
+
+    def test_commit_traversal(self):
+        # bound to cat-file parsing performance
+        nc = 0
+        st = time()
+        for c in self.gitrorepo.commit().traverse(branch_first=False):
+            nc += 1
+            self._query_commit_info(c)
+        # END for each traversed commit
+        elapsed_time = time() - st
+        print("Traversed %i Commits in %s [s] ( %f commits/s )"
+              % (nc, elapsed_time, nc / elapsed_time), file=sys.stderr)
+
+    def test_commit_iteration(self):
+        # bound to stream parsing performance
+        nc = 0
+        st = time()
+        for c in Commit.iter_items(self.gitrorepo, self.gitrorepo.head):
+            nc += 1
+            self._query_commit_info(c)
+        # END for each traversed commit
+        elapsed_time = time() - st
+        print("Iterated %i Commits in %s [s] ( %f commits/s )"
+              % (nc, elapsed_time, nc / elapsed_time), file=sys.stderr)
+
+    def test_commit_serialization(self):
+        self.assert_commit_serialization(self.gitrwrepo, '58c78e6', True)
+
+        rwrepo = self.gitrwrepo
+        make_object = rwrepo.odb.store
+        # direct serialization - deserialization can be tested afterwards
+        # serialization is probably limited on IO
+        hc = rwrepo.commit(rwrepo.head)
+
+        nc = 5000
+        st = time()
+        for i in range(nc):
+            cm = Commit(rwrepo, Commit.NULL_BIN_SHA, hc.tree,
+                        hc.author, hc.authored_date, hc.author_tz_offset,
+                        hc.committer, hc.committed_date, hc.committer_tz_offset,
+                        str(i), parents=hc.parents, encoding=hc.encoding)
+
+            stream = BytesIO()
+            cm._serialize(stream)
+            slen = stream.tell()
+            stream.seek(0)
+
+            cm.binsha = make_object(IStream(Commit.type, slen, stream)).binsha
+        # END commit creation
+        elapsed = time() - st
+
+        print("Serialized %i commits to loose objects in %f s ( %f commits / s )"
+              % (nc, elapsed, nc / elapsed), file=sys.stderr)