add tests

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)