Skip to content

Commit 9607c4f

Browse files
Add test of stream decompression speed
1 parent 5232d65 commit 9607c4f

File tree

2 files changed

+21
-4
lines changed

2 files changed

+21
-4
lines changed

tests/test_decompress.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ def test_decompression_null(self):
1111
else:
1212
DATA = b''
1313
self.assertRaises(zstd.Error, zstd.uncompress, zstd.compress(DATA)+b' ')
14-
14+
1515
def test_decompression_streamed(self):
1616
#log.info('cwd: %s' % os.getcwd())
1717
f = open("tests/test_data/facebook.ico.zst","rb")

tests/test_speed.py

Lines changed: 20 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ def test_compression_speed(self):
1919
cdata = zstd.compress(tDATA)
2020
sum+=l
2121

22-
log.info("Compression speed average =%6f Mb/sec" % (1.0*sum/1024/1024/wait,))
22+
log.info("Compression speed average = %6f Mb/sec" % (1.0*sum/1024/1024/wait,))
2323

24-
def test_decompression_speed(self):
24+
def test_decompression_block_speed(self):
2525
wait = 30
2626
log.info("\nWait %d seconds..." % wait)
2727
sum = 0
@@ -32,7 +32,24 @@ def test_decompression_speed(self):
3232
data = zstd.decompress(cdata)
3333
sum+=l
3434

35-
log.info("Decompression speed average =%6f Mb/sec" % (1.0*sum/1024/1024/wait,))
35+
log.info("Decompression of block data speed average = %6f Mb/sec" % (1.0*sum/1024/1024/wait,))
36+
37+
def test_decompression_stream_speed(self):
38+
wait = 30
39+
log.info("\nWait %d seconds..." % wait)
40+
sum = 0
41+
# cdata = zstd.compress(tDATA)
42+
# cdata = b'\x28\xb5\x2f\xfd\x00\x58\x11\x00\x00\x7b\x7d'
43+
f = open("tests/test_data/facebook.ico.zst","rb")
44+
cdata = f.read()
45+
f.close()
46+
l=len(cdata)
47+
tbegin = time()
48+
while time()-tbegin<wait:
49+
data = zstd.decompress(cdata)
50+
sum+=l
51+
52+
log.info("Decompression of streamed datspeed average = %6f Mb/sec" % (1.0*sum/1024/1024/wait,))
3653

3754
def test_check_speed(self):
3855
wait = 30

0 commit comments

Comments
 (0)