upstream/mercurial-mirror Files · contrib/python-zstandard/tests/test_cffi.py

util: expose an "available" API on compression engines...

util: expose an "available" API on compression engines When the zstd compression engine is introduced, it won't work in all installations, namely pure Python installs. So, we need a mechanism to declare whether a compression engine is available. We don't want to conditionally register the compression engine because it is sometimes useful to know when a compression engine name or encountered data is valid but just not available versus unknown.

Gregory Szorc - - Load All Authors

File last commit:

r30435:b86a448a default


                r30437:64d72754

default

Download file

             test_cffi.py
        
                    35 lines
            
             | 838 B
            
                | text/x-python
            
             |
                PythonLexer
            
             / contrib / python-zstandard / tests / test_cffi.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      import io

      try:

          import unittest2 as unittest

      except ImportError:

          import unittest

      import zstd

      try:

          import zstd_cffi

      except ImportError:

          raise unittest.SkipTest('cffi version of zstd not available')

      class TestCFFIWriteToToCDecompressor(unittest.TestCase):

          def test_simple(self):

              orig = io.BytesIO()

              orig.write(b'foo')

              orig.write(b'bar')

              orig.write(b'foobar' * 16384)

              dest = io.BytesIO()

              cctx = zstd_cffi.ZstdCompressor()

              with cctx.write_to(dest) as compressor:

                  compressor.write(orig.getvalue())

              uncompressed = io.BytesIO()

              dctx = zstd.ZstdDecompressor()

              with dctx.write_to(uncompressed) as decompressor:

                  decompressor.write(dest.getvalue())

              self.assertEqual(uncompressed.getvalue(), orig.getvalue())

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				import io

				try:
				import unittest2 as unittest
				except ImportError:
				import unittest

				import zstd

				try:
				import zstd_cffi
				except ImportError:
				raise unittest.SkipTest('cffi version of zstd not available')


				class TestCFFIWriteToToCDecompressor(unittest.TestCase):
				def test_simple(self):
				orig = io.BytesIO()
				orig.write(b'foo')
				orig.write(b'bar')
				orig.write(b'foobar' * 16384)

				dest = io.BytesIO()
				cctx = zstd_cffi.ZstdCompressor()
				with cctx.write_to(dest) as compressor:
				compressor.write(orig.getvalue())

				uncompressed = io.BytesIO()
				dctx = zstd.ZstdDecompressor()
				with dctx.write_to(uncompressed) as decompressor:
				decompressor.write(dest.getvalue())

				self.assertEqual(uncompressed.getvalue(), orig.getvalue())