summaryrefslogtreecommitdiff
path: root/test/deflate
diff options
context:
space:
mode:
Diffstat (limited to 'test/deflate')
-rwxr-xr-xtest/deflate18
1 files changed, 18 insertions, 0 deletions
diff --git a/test/deflate b/test/deflate
new file mode 100755
index 0000000..0271d59
--- /dev/null
+++ b/test/deflate
@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+
+import sys
+import zlib
+
+level = -1
+if len(sys.argv) > 2:
+ level = int(sys.argv[2])
+
+try:
+ with open(sys.argv[1], "rb") as f:
+ input_data = f.read()
+except FileNotFoundError:
+ input_data = sys.argv[1].encode("utf-8")
+
+output = zlib.compress(input_data, level=level)
+
+sys.stdout.buffer.write(output)