test/test_basic.rb in extzstd-0.1.1 vs test/test_basic.rb in extzstd-0.2

@@ -30,10 +30,18 @@
     d = StringIO.new("")
     Zstd.encode(d) { |z| 1000.times { z << src; size_a += src.bytesize; md5a.update src } }
     d.pos = 0
     size_b = 0
     md5b = Digest::MD5.new
-    Zstd.decode(d) { |z| buf = ""; while z.read(654321, buf); size_b += buf.bytesize; md5b.update buf; end }
+    partial_list = [262144, 1, 262144, 262142, 524288, 524280, 99, 999, 9999, 99999, 999999, 9999999, nil]
+    Zstd.decode(d) do |z|
+      buf = ""
+      while z.read(s = partial_list.shift, buf)
+        assert_equal s, buf.bytesize if s
+        size_b += buf.bytesize
+        md5b.update buf
+      end
+    end
     assert_equal size_a, size_b
     assert_equal(md5a.hexdigest, md5b.hexdigest)
   end
 
   def test_dictionary