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

util: check for compression engine availability before returning...

util: check for compression engine availability before returning If a requested compression engine is registered but not available, requesting it will now abort. To be honest, I'm not sure if this is the appropriate mechanism for handling optional compression engines. I won't know until all uses of compression (bundles, wire protocol, revlogs, etc) are using the new API and zstd (our planned optional engine) is implemented. So this API could change.

Gregory Szorc - - Load All Authors

File last commit:

r30435:b86a448a default


                r30438:90933e4e

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())