def checkResult(self, data, sorted_data, compare=None):
barrier()
all_data = get_comm_world().gather(data, root=0)
all_sorted_data = get_comm_world().gather(sorted_data, root=0)
if is_rank_zero():
all_data = flatten_list(all_data)
sort_list(all_data, compare)
all_sorted_data = flatten_list(all_sorted_data)
areEqual = all_data == all_sorted_data
else:
areEqual = None
areEqual = get_comm_world().bcast(areEqual, 0)
self.assertTrue(areEqual)
def main():
args = parse_command_line()
check_command_line(args)
time_steps = get_time_steps(args)
check_time_step_files(args, time_steps)
grid_desc = create_grid_description(args)
program_data = distribute_program_data(args, grid_desc, time_steps)
(unstructured_grid, global_ids) = create_grid(program_data)
write_grid(unstructured_grid, global_ids, program_data)
if is_rank_zero():
write_pvd_file(unstructured_grid, program_data)
barrier()
if is_rank_zero():
print("grid2paraview_cells finished")
def testSpartaCircleGridFileSortToFileBuckets(self):
barrier()
self.sortGridFileToFileBuckets(CIRCLE_GRID_FILE)
def testSpartaGridFile200SortToFileBuckets(self):
barrier()
self.sortGridFileToFileBuckets(GRID_FILE_200)
def testSpartaCircleGridFile(self):
barrier()
self.sortGridFile(CIRCLE_GRID_FILE)
self.sortGridFile(CIRCLE_GRID_FILE, use_file_buckets=True)
def testSpartaGridFile200(self):
barrier()
self.sortGridFile(GRID_FILE_200)
self.sortGridFile(GRID_FILE_200, use_file_buckets=True)
def testSortSize1000000(self):
barrier()
data = self.generateData(1000000)
self.checkResult(data, parallel_sort(data))
def testSortCompareInput(self):
barrier()
with self.assertRaises(ValueError):
parallel_sort([], 8.2)
def testSortDataInput(self):
barrier()
with self.assertRaises(ValueError):
parallel_sort(5)
