def test():
comm=PSim(5,SWITCH)
if comm.rank==0: print 'start test'
a=sum(comm.all2all_broadcast(comm.rank))
comm.barrier()
b=comm.all2all_reduce(comm.rank)
if a!=10 or a!=b:
print 'from process', comm.rank
raise Exception
if comm.rank==0: print 'test passed'
def test():
comm = PSim(5, SWITCH)
if comm.rank == 0: print 'start test'
a = sum(comm.all2all_broadcast(comm.rank))
comm.barrier()
b = comm.all2all_reduce(comm.rank)
if a != 10 or a != b:
print 'from process', comm.rank
raise Exception
if comm.rank == 0: print 'test passed'
