Thanks to visit codestin.com
Credit goes to github.com

Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
restore out-of-bounds tests & improve range notation
  • Loading branch information
AA-Turner committed May 28, 2025
commit 62d085e50b7d12f181cb6d3b982149dc29ae2422
19 changes: 13 additions & 6 deletions Lib/test/test_zstd.py
Original file line number Diff line number Diff line change
Expand Up @@ -200,16 +200,16 @@ def test_simple_compress_bad_args(self):
self.assertRaises(TypeError, ZstdCompressor, zstd_dict={1: 2, 3: 4})

# valid range for compression level is [-(1<<17), 22]
msg = 'compression level {} not in valid range -131072 <= level <= 22'
msg = 'illegal compression level {}; the valid range is [-131072, 22]'
with self.assertRaisesRegex(ValueError, msg.format(C_INT_MAX)):
ZstdCompressor(C_INT_MAX)
with self.assertRaisesRegex(ValueError, msg.format(C_INT_MIN)):
ZstdCompressor(C_INT_MIN)
msg = 'compression level not in valid range -131072 <= level <= 22'
msg = 'illegal compression level; the valid range is [-131072, 22]'
with self.assertRaisesRegex(ValueError, msg):
ZstdCompressor(level=-(2**1000))
with self.assertRaisesRegex(ValueError, msg):
ZstdCompressor(level=(2**1000))
ZstdCompressor(level=2**1000)

with self.assertRaises(ValueError):
ZstdCompressor(options={CompressionParameter.window_log: 100})
Expand Down Expand Up @@ -305,7 +305,7 @@ def test_compress_parameters(self):
with self.assertRaisesRegex(
ValueError,
"compression parameter 'window_log' received an illegal value 100; "
r'the valid range is \d+ <= value <= \d+',
r'the valid range is \[\d+, \d+\]',
):
compress(b'', options=option)

Expand Down Expand Up @@ -439,10 +439,17 @@ def test_decompress_parameters(self):

d1 = d.copy()
# larger than signed int
d1[DecompressionParameter.window_log_max] = 2**1000
with self.assertRaises(OverflowError):
ZstdDecompressor(None, d1)
# smaller than signed int
d1[DecompressionParameter.window_log_max] = -(2**1000)
with self.assertRaises(OverflowError):
ZstdDecompressor(None, d1)

d1[DecompressionParameter.window_log_max] = C_INT_MAX
with self.assertRaises(ValueError):
ZstdDecompressor(None, d1)
# smaller than signed int
d1[DecompressionParameter.window_log_max] = C_INT_MIN
with self.assertRaises(ValueError):
ZstdDecompressor(None, d1)
Expand All @@ -452,7 +459,7 @@ def test_decompress_parameters(self):
with self.assertRaisesRegex(
ValueError,
"decompression parameter 'window_log_max' received an illegal value 100; "
r'the valid range is \d+ <= value <= \d+',
r'the valid range is \[\d+, \d+\]',
):
decompress(b'', options=options)

Expand Down
2 changes: 1 addition & 1 deletion Modules/_zstd/_zstdmodule.c
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ set_parameter_error(int is_compress, int key_v, int value_v)
/* Error message */
PyErr_Format(PyExc_ValueError,
"%s parameter '%s' received an illegal value %d; "
"the valid range is %d <= value <= %d",
"the valid range is [%d, %d]",
type, name, value_v, bounds.lowerBound, bounds.upperBound);
}

Expand Down
4 changes: 2 additions & 2 deletions Modules/_zstd/compressor.c
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ _zstd_set_c_level(ZstdCompressor *self, int level)
int max_level = ZSTD_maxCLevel();
if (level < min_level || level > max_level) {
PyErr_Format(PyExc_ValueError,
"compression level %d not in valid range %d <= level <= %d",
"illegal compression level %d; the valid range is [%d, %d]",
level, min_level, max_level);
return -1;
}
Expand Down Expand Up @@ -378,7 +378,7 @@ _zstd_ZstdCompressor_new_impl(PyTypeObject *type, PyObject *level,
if (level_v == -1 && PyErr_Occurred()) {
if (PyErr_ExceptionMatches(PyExc_OverflowError)) {
PyErr_Format(PyExc_ValueError,
"compression level not in valid range %d <= level <= %d",
"illegal compression level; the valid range is [%d, %d]",
ZSTD_minCLevel(), ZSTD_maxCLevel());
}
goto error;
Expand Down