1 files changed, 12 insertions, 9 deletions
diff --git a/gitdb/test/performance/test_pack_streaming.py b/gitdb/test/performance/test_pack_streaming.py
index c66e60c..fe160ea 100644
--- a/gitdb/test/performance/test_pack_streaming.py
+++ b/gitdb/test/performance/test_pack_streaming.py
@@ -3,18 +3,20 @@
 # This module is part of GitDB and is released under
 # the New BSD License: http://www.opensource.org/licenses/bsd-license.php
 """Specific test for pack streams only"""
-from lib import (
+from __future__ import print_function
+
+from gitdb.test.performance.lib import (
     TestBigRepoR 
-    )
+)
 
 from gitdb.db.pack import PackedDB
 from gitdb.stream import NullStream
 from gitdb.pack import PackEntity
+from gitdb.test.lib import skip_on_travis_ci
 
 import os
 import sys
 from time import time
-from nose import SkipTest
 
 class CountedNullStream(NullStream):
     __slots__ = '_bw'
@@ -30,15 +32,15 @@ class CountedNullStream(NullStream):
 
 class TestPackStreamingPerformance(TestBigRepoR):
     
+    @skip_on_travis_ci
     def test_pack_writing(self):
         # see how fast we can write a pack from object streams.
         # This will not be fast, as we take time for decompressing the streams as well
         ostream = CountedNullStream()
         pdb = PackedDB(os.path.join(self.gitrepopath, "objects/pack"))
         
-        ni = 5000
+        ni = 1000
         count = 0
-        total_size = 0
         st = time()
         for sha in pdb.sha_iter():
             count += 1
@@ -47,17 +49,18 @@ class TestPackStreamingPerformance(TestBigRepoR):
                 break
         #END gather objects for pack-writing
         elapsed = time() - st
-        print >> sys.stderr, "PDB Streaming: Got %i streams by sha in in %f s ( %f streams/s )" % (ni, elapsed, ni / elapsed)
+        print("PDB Streaming: Got %i streams by sha in in %f s ( %f streams/s )" % (ni, elapsed, ni / elapsed), file=sys.stderr)
         
         st = time()
         PackEntity.write_pack((pdb.stream(sha) for sha in pdb.sha_iter()), ostream.write, object_count=ni)
         elapsed = time() - st
         total_kb = ostream.bytes_written() / 1000
-        print >> sys.stderr, "PDB Streaming: Wrote pack of size %i kb in %f s (%f kb/s)" % (total_kb, elapsed, total_kb/elapsed)
+        print(sys.stderr, "PDB Streaming: Wrote pack of size %i kb in %f s (%f kb/s)" % (total_kb, elapsed, total_kb/elapsed), sys.stderr)
         
     
+    @skip_on_travis_ci
     def test_stream_reading(self):
-        raise SkipTest()
+        # raise SkipTest()
         pdb = PackedDB(os.path.join(self.gitrepopath, "objects/pack"))
         
         # streaming only, meant for --with-profile runs
@@ -75,5 +78,5 @@ class TestPackStreamingPerformance(TestBigRepoR):
             count += 1
         elapsed = time() - st
         total_kib = total_size / 1000
-        print >> sys.stderr, "PDB Streaming: Got %i streams by sha and read all bytes totallying %i KiB ( %f KiB / s ) in %f s ( %f streams/s )" % (ni, total_kib, total_kib/elapsed , elapsed, ni / elapsed)
+        print(sys.stderr, "PDB Streaming: Got %i streams by sha and read all bytes totallying %i KiB ( %f KiB / s ) in %f s ( %f streams/s )" % (ni, total_kib, total_kib/elapsed , elapsed, ni / elapsed), sys.stderr)