code: def test_fits_within_context_window(self):
      # Should fit in the context window since we subtracted the number of tokens of the prompt
      # from the max context window.
      assert self.window_service.fits_within_context_window(
          self.prompt, self.window_service.max_request_length - 6
      )
      # 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(
          self.prompt, self.window_service.max_request_length - 6 + 1
      )
  
file_path: H2O/h2o_hf/helm/src/helm/benchmark/window_services/test_cohere_window_service.py
imports: import os
  import shutil
  import tempfile
  from typing import List
  from sqlitedict import SqliteDict
  from helm.common.general import ensure_directory_exists
  from .test_cohere_window_service_utils import REQUESTS_TO_RESPONSES, TEST_PROMPT, TOKENIZED_PROMPT
  from .tokenizer_service import TokenizerService
  from .window_service_factory import WindowServiceFactory
  from .test_utils import get_tokenizer_service
label_desc: Use image node_id 8 for calling the TestCohereWindowService obj's underlying member method code with example usage: obj.test_fits_within_context_window() without return types