def METHOD_NAME(self):
    # Should fit in the context window since we subtracted the number of tokens of the test prompt
    # from the max context window
    assert self.window_service.fits_within_context_window(
        TEST_PROMPT, self.window_service.max_request_length - self.TEST_PROMPT_LENGTH
    )
    # Should not fit in the context window because we're expecting one more extra token in the completion
    assert not self.window_service.fits_within_context_window(
        TEST_PROMPT, self.window_service.max_request_length - self.TEST_PROMPT_LENGTH + 1
    )