update bench.py
add assert to dropAndReshard