def test_with_usage(self):
        """Ensures token counts are extracted correctly when usage is present."""
        raw_response = {
            "usage": {
                "prompt_tokens": 10,
                "completion_tokens": 5,
                "total_tokens": 15,
            }
        }
        expected_result = {
            "prompt_tokens": 10,
            "completion_tokens": 5,
            "total_tokens": 15,
        }
        result = _get_response_token_counts(raw_response)
        assert result == expected_result
Ensures token counts are extracted correctly when usage is present.
llama-index-integrations/llms/llama-index-llms-oci-data-science/tests/test_oci_data_science_utils.py