Added test case

heuer · Aug 16, 2016 · 7173587 · 7173587
1 parent 6c331b3
commit 7173587
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/README.rst b/README.rst
@@ -8,7 +8,7 @@ This package implements main parts of ISO/IEC 18004:2006(E) / ISO/IEC 18004:2015
 and produces Micro QR Codes and QR Codes with nearly no effort.
 
 Segno provides several serialization formats like SVG, EPS, PNG, PDF, or text
-output. Neither of these serializers require an external lib. Segno provides
+output. None of these serializers require an external lib. Segno provides
 more serialization formats via a plugin architecture. It was tested against
 Python 2.6 - 3.4 and PyPy.
 

diff --git a/tests/test_qrcode.py b/tests/test_qrcode.py
@@ -236,6 +236,16 @@ def test_neq():
     ok_(qr != qr2)
 
 
+def test_save_png_buffer():
+    qr = segno.make_qr('test')
+    out = io.BytesIO()
+    qr.save(out, kind='png')
+    out.seek(0)
+    expected = b'\211PNG\r\n\032\n'  # PNG magic number
+    val = out.read(len(expected))
+    eq_(expected, val)
+
+
 def test_save_png_filestream():
     qr = segno.make_qr('test')
     f = tempfile.NamedTemporaryFile('wb', suffix='.png', delete=False)