1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# (C) Copyright 2005-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
"""
Tests for unpicking of pickles created using previous versions
of Traits.
"""
import pathlib
import pickle
import unittest
from traits.testing.optional_dependencies import (
pkg_resources,
requires_pkg_resources,
)
def find_pickles():
"""
Iterate over the pickle files in the test_data directory.
Skip files that correspond to a protocol not supported with
the current version of Python.
Yields paths to pickle files.
"""
pickle_directory = pathlib.Path(
pkg_resources.resource_filename(
"traits.tests", "test-data/historical-pickles",
)
)
for pickle_path in pickle_directory.glob("*.pkl"):
header, _, protocol, _ = pickle_path.name.split("-", maxsplit=3)
if header != "hipt":
# Skip pickle files that don't follow the naming convention.
continue
if not protocol.startswith("p"):
raise RuntimeError("Can't interpret protocol: {}".format(protocol))
protocol = int(protocol[1:])
if protocol > pickle.HIGHEST_PROTOCOL:
# Protocol not understood by current Python; skip.
continue
yield pickle_path
class TestHistoricalPickles(unittest.TestCase):
@requires_pkg_resources
def test_unpickling_historical_pickles(self):
# Just test that the pickle can be unpickled.
for pickle_path in find_pickles():
with self.subTest(filename=pickle_path.name):
with pickle_path.open("rb") as f:
pickle.load(f)
|
