Fix tests on py2

This commit is contained in:
Antoine Pitrou committed 2018-08-13 19:19:23 +02:00
1 parent 42f01864f2
commit 06c0f5f4ec
1 file changed
+2 -2
+2 -2
View File
@@ -153,8 +153,8 @@ def check_example_using_pickle(example_name):
JSON files.
"""
pickle_path = path_for_pickle_example(example_name)
with bz2.open(pickle_path, 'rb') as f:
expected_cols = pickle.load(f)
with open(pickle_path, 'rb') as f:
expected_cols = pickle.loads(bz2.decompress(f.read()))
check_example_file(path_for_orc_example(example_name), expected_cols)