Merge "Relax assertion in test_get_all_shards"