code: def test_enough_gpus() -> None:
      # not enough gpus
      try:
          LinuxTesterContainer(
              "team", shard_count=2, gpus=1, skip_ray_installation=True
          )
      except AssertionError:
          pass
      else:
          assert False, "Should raise an AssertionError"
  
      # not enough gpus
      try:
          LinuxTesterContainer(
              "team", shard_count=1, gpus=1, skip_ray_installation=True
          )
      except AssertionError:
          assert False, "Should not raise an AssertionError"
  
imports: import json
  import os
  import platform
  import sys
  import pytest
  import tempfile
  from unittest import mock
  from typing import List, Optional
  from ci.ray_ci.linux_tester_container import LinuxTesterContainer
  from ci.ray_ci.utils import chunk_into_n
  from ci.ray_ci.container import _DOCKER_ECR_REPO, _RAYCI_BUILD_ID
label_desc: Use image node_id 1 for calling a global function with example usage: test_enough_gpus() without return types