fix test

Signed-off-by: Future-Outlier <eric901201@gmail.com>

tests/flytekit/unit/core/test_container_task.py

@@ -23,37 +23,33 @@
 
 def test_local_execution():
  try:
- import docker
-
- docker.from_env()
+ calculate_ellipse_area_python = ContainerTask(
+ name="calculate_ellipse_area_python",
+ input_data_dir="/var/inputs",
+ output_data_dir="/var/outputs",
+ inputs=kwtypes(a=float, b=float),
+ outputs=kwtypes(area=float, metadata=str),
+ image="ghcr.io/flyteorg/rawcontainers-python:v2",
+ command=[
+ "python",
+ "calculate-ellipse-area.py",
+ "{{.inputs.a}}",
+ "{{.inputs.b}}",
+ "/var/outputs",
+ ],
+ )
+
+ def wf() -> Tuple[float, str]:
+ return calculate_ellipse_area_python(a=3.0, b=4.0)
+
+ area, metadata = wf()
+ assert area == 37.69911184307752
+ assert metadata == "[from python rawcontainer]"
  except Exception as e:
  # Currently, Ubuntu will pass the test, but MacOS and Windows will not
  print(f"Skipping test due to Docker environment setup failure: {e}")
  return
 
- calculate_ellipse_area_python = ContainerTask(
- name="calculate_ellipse_area_python",
- input_data_dir="/var/inputs",
- output_data_dir="/var/outputs",
- inputs=kwtypes(a=float, b=float),
- outputs=kwtypes(area=float, metadata=str),
- image="ghcr.io/flyteorg/rawcontainers-python:v2",
- command=[
- "python",
- "calculate-ellipse-area.py",
- "{{.inputs.a}}",
- "{{.inputs.b}}",
- "/var/outputs",
- ],
- )
-
- def wf() -> Tuple[float, str]:
- return calculate_ellipse_area_python(a=3.0, b=4.0)
-
- area, metadata = wf()
- assert area == 37.69911184307752
- assert metadata == "[from python rawcontainer]"
-
 
 def test_pod_template():
  ps = V1PodSpec(