updated federated partition provider tests