change number of partitions during tests