Fix pyarrow version (#3760)

This commit is contained in:
Philipp Moritz
2019-01-13 14:28:23 -08:00
committed by Robert Nishihara
parent 3adffe6a4e
commit 00e9f8d870
5 changed files with 21 additions and 2 deletions
+15
View File
@@ -8,9 +8,11 @@ import os
import random
import re
import setproctitle
import shutil
import string
import subprocess
import sys
import tempfile
import threading
import time
from collections import defaultdict, namedtuple, OrderedDict
@@ -2585,3 +2587,16 @@ def test_ray_stack(shutdown_only):
if not success:
raise Exception("Failed to find necessary information with "
"'ray stack'")
def test_pandas_parquet_serialization():
# Only test this if pandas is installed
pytest.importorskip("pandas")
import pandas as pd
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, "parquet-test")
pd.DataFrame({"col1": [0, 1], "col2": [0, 1]}).to_parquet(filename)
# Clean up
shutil.rmtree(tempdir)