Code review comment: Can we also add tests with container with multi-partitions?

Code under review:
public void before_NonStreamingOrderByQueryVectorSearchTest() {
        
        client = new CosmosClientBuilder()
            .endpoint(TestConfigurations.HOST)
            .key(TestConfigurations.MASTER_KEY)
            .directMode(DirectConnectionConfig.getDefaultConfig())
            .consistencyLevel(ConsistencyLevel.SESSION)
            .contentResponseOnWriteEnabled(true)
            .buildAsyncClient();

        database = createDatabase(client, databaseId);

        PartitionKeyDefinition partitionKeyDef = new PartitionKeyDefinition();
        ArrayList<String> paths = new ArrayList<String>();
        paths.add("/pk");
        partitionKeyDef.setPaths(paths);

        CosmosContainerProperties containerProperties = new CosmosContainerProperties(flatContainerId, partitionKeyDef);
        containerProperties.setIndexingPolicy(populateIndexingPolicy(CosmosVectorIndexType.FLAT));
        containerProperties.setVectorEmbeddingPolicy(populateVectorEmbeddingPolicy(128L));
        database.createContainer(containerProperties).block();
        flatIndexContainer = database.getContainer(flatContainerId);

        containerProperties = new CosmosContainerProperties(quantizedContainerId, partitionKeyDef);
        containerProperties.setIndexingPolicy(populateIndexingPolicy(CosmosVectorIndexType.QUANTIZED_FLAT));
        containerProperties.setVectorEmbeddingPolicy(populateVectorEmbeddingPolicy(128L));
        database.createContainer(containerProperties).block();
        quantizedIndexContainer = database.getContainer(quantizedContainerId);

        containerProperties = new CosmosContainerProperties(largeDataContainerId, partitionKeyDef);
        containerProperties.setIndexingPolicy(populateIndexingPolicy(CosmosVectorIndexType.QUANTIZED_FLAT));
        containerProperties.setVectorEmbeddingPolicy(populateVectorEmbeddingPolicy(2L));
        database.createContainer(containerProperties).block();
        largeDataContainer = database.getContainer(largeDataContainerId);

        for (Document doc: getVectorDocs()) {
            flatIndexContainer.createItem(doc).block();
            quantizedIndexContainer.createItem(doc).block();
        }
    }

Line commented on: database.createContainer(containerProperties).block();

Revised code:
public void before_NonStreamingOrderByQueryVectorSearchTest() {
        
        client = new CosmosClientBuilder()
            .endpoint(TestConfigurations.HOST)
            .key(TestConfigurations.MASTER_KEY)
            .directMode(DirectConnectionConfig.getDefaultConfig())
            .consistencyLevel(ConsistencyLevel.SESSION)
            .contentResponseOnWriteEnabled(true)
            .buildAsyncClient();

        database = createDatabase(client, databaseId);

        PartitionKeyDefinition partitionKeyDef = new PartitionKeyDefinition();
        ArrayList<String> paths = new ArrayList<String>();
        paths.add("/pk");
        partitionKeyDef.setPaths(paths);

        CosmosContainerProperties containerProperties = new CosmosContainerProperties(flatContainerId, partitionKeyDef);
        containerProperties.setIndexingPolicy(populateIndexingPolicy(CosmosVectorIndexType.FLAT));
        containerProperties.setVectorEmbeddingPolicy(populateVectorEmbeddingPolicy(128L));
        database.createContainer(containerProperties).block();
        flatIndexContainer = database.getContainer(flatContainerId);

        containerProperties = new CosmosContainerProperties(quantizedContainerId, partitionKeyDef);
        containerProperties.setIndexingPolicy(populateIndexingPolicy(CosmosVectorIndexType.QUANTIZED_FLAT));
        containerProperties.setVectorEmbeddingPolicy(populateVectorEmbeddingPolicy(128L));
        database.createContainer(containerProperties, ThroughputProperties.createManualThroughput(20000)).block();
        quantizedIndexContainer = database.getContainer(quantizedContainerId);

        containerProperties = new CosmosContainerProperties(largeDataContainerId, partitionKeyDef);
        containerProperties.setIndexingPolicy(populateIndexingPolicy(CosmosVectorIndexType.QUANTIZED_FLAT));
        containerProperties.setVectorEmbeddingPolicy(populateVectorEmbeddingPolicy(2L));
        database.createContainer(containerProperties).block();
        largeDataContainer = database.getContainer(largeDataContainerId);

        for (Document doc : getVectorDocs()) {
            flatIndexContainer.createItem(doc).block();
            quantizedIndexContainer.createItem(doc).block();
        }
    }