def METHOD_NAME(self) -> None:
    cases = [
        {
            "batch_size": BATCH_SIZE - 1,
            "expected_number_of_requests": 1,
        },
        {
            "batch_size": BATCH_SIZE,
            "expected_number_of_requests": 1,
        },
        {
            "batch_size": BATCH_SIZE + 1,
            "expected_number_of_requests": 2,
        },
    ]
    for case in cases:
        mock_client = mock.Mock()
        mock_client.Documents.add_documents.return_value = "OK"
        batch_size = case["batch_size"]
        expected_number_of_requests = case["expected_number_of_requests"]
        docs = list(range(batch_size))
        upload_to_rockset(
            collection="test", docs=docs, workspace="commons", client=mock_client
        )
        self.assertEqual(
            mock_client.Documents.add_documents.call_count,
            expected_number_of_requests,
        )