From 5f8e0dca01e74a8df45c86fb0fddf177a5b2a113 Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Thu, 4 Nov 2021 12:44:19 -0700 Subject: [PATCH] add 11.5 xfail --- python/cudf/cudf/tests/test_parquet.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/python/cudf/cudf/tests/test_parquet.py b/python/cudf/cudf/tests/test_parquet.py index 45cf56353a9..9629e502584 100644 --- a/python/cudf/cudf/tests/test_parquet.py +++ b/python/cudf/cudf/tests/test_parquet.py @@ -17,6 +17,8 @@ from packaging import version from pyarrow import fs as pa_fs, parquet as pq +import rmm + import cudf from cudf.io.parquet import ParquetWriter, merge_parquet_filemetadata from cudf.testing import dataset_generator as dg @@ -172,7 +174,9 @@ def gdf_day_timestamps(pdf_day_timestamps): @pytest.fixture(params=["snappy", "gzip", "brotli", None, np.str_("snappy")]) def parquet_file(request, tmp_path_factory, pdf): - fname = tmp_path_factory.mktemp("parquet") / "test.parquet" + fname = tmp_path_factory.mktemp("parquet") / ( + str(request.param) + "_test.parquet" + ) pdf.to_parquet(fname, engine="pyarrow", compression=request.param) return fname @@ -242,6 +246,12 @@ def _make_parquet_path_or_buf(src): ], ) def test_parquet_reader_basic(parquet_file, columns, engine): + if rmm._cuda.gpu.runtimeGetVersion() == 11050 and "brotli_" in str( + parquet_file + ): + pytest.xfail( + "Known issue: https://github.com/rapidsai/cudf/issues/9546" + ) expect = pd.read_parquet(parquet_file, columns=columns) got = cudf.read_parquet(parquet_file, engine=engine, columns=columns) if len(expect) == 0: