diff --git a/Lib/test/test_collections.py b/Lib/test/test_collections.py index 56e81207058..ee28a6c0b35 100644 --- a/Lib/test/test_collections.py +++ b/Lib/test/test_collections.py @@ -301,7 +301,7 @@ def test_pickle(self): for module in (pickle,): loads = getattr(module, 'loads') dumps = getattr(module, 'dumps') - for protocol in -1, 0, 1, 2: + for protocol in range(-1, module.HIGHEST_PROTOCOL + 1): q = loads(dumps(p, protocol)) self.assertEqual(p, q) self.assertEqual(p._fields, q._fields)