add assert for partition
fix tests to always pass a partition