Skip to content

Commit f923c97

Browse files
authored
BSE-4706: Fix test tmp_path sync (#382)
1 parent 740d581 commit f923c97

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

bodo/tests/test_parquet_write.py

+6
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,8 @@ def check_write_func(
158158
),
159159
)
160160
def test_semi_structured_data(df, tmp_path):
161+
comm = bodo.mpi4py.MPI.COMM_WORLD
162+
tmp_path = comm.bcast(tmp_path if bodo.get_rank() == 0 else None, root=0)
161163
check_write_func(
162164
lambda df, path: df.to_parquet(path),
163165
df,
@@ -1086,6 +1088,8 @@ def impl(df, path):
10861088
def test_to_pq_multiIdx(tmp_path, memory_leak_check):
10871089
"""Test to_parquet with MultiIndexType"""
10881090
np.random.seed(0)
1091+
comm = bodo.mpi4py.MPI.COMM_WORLD
1092+
tmp_path = comm.bcast(tmp_path if bodo.get_rank() == 0 else None, root=0)
10891093

10901094
arrays = [
10911095
["bar", "bar", "baz", "baz", "foo", "foo", "qux", "qux"],
@@ -1108,6 +1112,8 @@ def test_to_pq_multiIdx(tmp_path, memory_leak_check):
11081112
def test_to_pq_multiIdx_no_name(tmp_path, memory_leak_check):
11091113
"""Test to_parquet with MultiIndexType with no name at 1 level"""
11101114
np.random.seed(0)
1115+
comm = bodo.mpi4py.MPI.COMM_WORLD
1116+
tmp_path = comm.bcast(tmp_path if bodo.get_rank() == 0 else None, root=0)
11111117

11121118
arrays = [
11131119
["bar", "bar", "baz", "baz", "foo", "foo", "qux", "qux"],

0 commit comments

Comments
 (0)