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

shelve: move patch extension to a string constant...

shelve: move patch extension to a string constant We are using 'name + ".patch"' pattern throughout the shelve code to identify the existence of a shelve with a particular name. In two cases however we use 'name + ".hg"' instead. This commit makes 'patch' be used in all places and "emphasizes" it by moving 'patch' to live in a constant. Also, this allows to extract file name without extension like this: f[:-(1 + len(patchextension))] instead of: f[:-6] which is good IMO. This is a first patch from this initial "obsshelve" series. This series does not include tests, although locally I have all of test-shelve.t ported to test obs-shelve as well. I will send tests later as a separate series.

Gregory Szorc - - Load All Authors

File last commit:

r30435:b86a448a default


                r30554:1775975d

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