code: def test_get_image_name(self) -> None:
      v = DEFAULT_PYTHON_VERSION
      pv = self.get_python_version(v)
      container = RayDockerContainer(v, "cpu", "ray")
      assert container._get_image_names() == [
          f"rayproject/ray:123456-{pv}-cpu",
          "rayproject/ray:123456-cpu",
          f"rayproject/ray:123456-{pv}",
          "rayproject/ray:123456",
          f"rayproject/ray:nightly-{pv}-cpu",
          "rayproject/ray:nightly-cpu",
          f"rayproject/ray:nightly-{pv}",
          "rayproject/ray:nightly",
      ]
  
      v = self.get_non_default_python()
      pv = self.get_python_version(v)
      container = RayDockerContainer(v, "cu11.8.0", "ray-ml")
      assert container._get_image_names() == [
          f"rayproject/ray-ml:123456-{pv}-cu118",
          f"rayproject/ray-ml:123456-{pv}-gpu",
          f"rayproject/ray-ml:123456-{pv}",
          f"rayproject/ray-ml:nightly-{pv}-cu118",
          f"rayproject/ray-ml:nightly-{pv}-gpu",
          f"rayproject/ray-ml:nightly-{pv}",
      ]
  
      with mock.patch.dict(
          os.environ, {"BUILDKITE_BRANCH": "releases/1.0.0"}
      ):
          v = DEFAULT_PYTHON_VERSION
          pv = self.get_python_version(v)
          container = RayDockerContainer(v, "cpu", "ray")
          assert container._get_image_names() == [
              f"rayproject/ray:1.0.0.123456-{pv}-cpu",
              "rayproject/ray:1.0.0.123456-cpu",
              f"rayproject/ray:1.0.0.123456-{pv}",
              "rayproject/ray:1.0.0.123456",
          ]
  
file_path: ray/ci/ray_ci/test_ray_docker_container.py
imports: import os
  import sys
  from typing import List
  from unittest import mock
  import pytest
  from ci.ray_ci.builder_container import DEFAULT_PYTHON_VERSION
  from ci.ray_ci.container import _DOCKER_ECR_REPO
  from ci.ray_ci.ray_docker_container import RayDockerContainer
  from ci.ray_ci.test_base import RayCITestBase
  from ci.ray_ci.utils import RAY_VERSION
label_desc: Use image node_id 4 for calling the TestRayDockerContainer obj's underlying member method code with example usage: obj.test_get_image_name() without return types