Create balanced_partition_test.rb