code: def test_fits_within_context_window(self, mock_tokenize):
      # 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, 2047 - 36
      )
      # 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, 2047 - 36 + 1
      )
  
file_path: H2O/h2o_hf/helm/src/helm/benchmark/window_services/test_ai21_window_service.py
imports: import pytest
  from typing import List
  import unittest
  from unittest import mock
  from helm.common.authentication import Authentication
  from helm.common.tokenization_request import TokenizationRequestResult, TokenizationToken, TextRange
  from helm.proxy.services.remote_service import RemoteService
  from .test_utils import TEST_PROMPT
  from .tokenizer_service import TokenizerService
  from .window_service_factory import WindowServiceFactory
label_desc: Use image node_id 5 for calling the TestAI21WindowService obj's underlying member method code with example usage: obj.test_fits_within_context_window(mock_tokenize) without return types