add testing with partitioning included