Fix failing test in planted_partition_graph

Update size to 218