@@ -158,6 +158,8 @@ def check_write_func(
158
158
),
159
159
)
160
160
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 )
161
163
check_write_func (
162
164
lambda df , path : df .to_parquet (path ),
163
165
df ,
@@ -1086,6 +1088,8 @@ def impl(df, path):
1086
1088
def test_to_pq_multiIdx (tmp_path , memory_leak_check ):
1087
1089
"""Test to_parquet with MultiIndexType"""
1088
1090
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 )
1089
1093
1090
1094
arrays = [
1091
1095
["bar" , "bar" , "baz" , "baz" , "foo" , "foo" , "qux" , "qux" ],
@@ -1108,6 +1112,8 @@ def test_to_pq_multiIdx(tmp_path, memory_leak_check):
1108
1112
def test_to_pq_multiIdx_no_name (tmp_path , memory_leak_check ):
1109
1113
"""Test to_parquet with MultiIndexType with no name at 1 level"""
1110
1114
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 )
1111
1117
1112
1118
arrays = [
1113
1119
["bar" , "bar" , "baz" , "baz" , "foo" , "foo" , "qux" , "qux" ],
0 commit comments