view contrib/python-zstandard/tests/test_estimate_sizes.py @ 30830:08fa3a76a080

zstd: prevent potential free() of uninitialized memory This is a cherry pick of an upstream fix. The free() of uninitialed memory could likely only occur if a malloc() inside zstd fails. The patched functions aren't currently used by Mercurial. But I don't like leaving footguns sitting around.
author Gregory Szorc <gregory.szorc@gmail.com>
date Tue, 17 Jan 2017 10:17:13 -0800
parents b86a448a2965
children c32454d69b85
line wrap: on
line source

try:
    import unittest2 as unittest
except ImportError:
    import unittest

import zstd


class TestSizes(unittest.TestCase):
    def test_decompression_size(self):
        size = zstd.estimate_decompression_context_size()
        self.assertGreater(size, 100000)

    def test_compression_size(self):
        params = zstd.get_compression_parameters(3)
        size = zstd.estimate_compression_context_size(params)
        self.assertGreater(size, 100000)