code: def test_shard_option():
      # for shard_option in [ShardOption.STANDARD, ShardOption.SHARD, ShardOption.FULL_SHARD, ShardOption.SHARD_LAST_AXIS]:
      for shard_option in [ShardOption.SHARD_LAST_AXIS]:
          check_shard_option(shard_option)
  
file_path: ColossalAI/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_shard_option.py
imports: import torch
  import torch.nn
  from colossalai._analyzer.fx.graph_module import ColoGraphModule
  from colossalai._analyzer.fx.passes.shape_prop import shape_prop_pass
  from colossalai._analyzer.fx.tracer.tracer import ColoTracer
  from colossalai.auto_parallel.tensor_shard.node_handler import LinearFunctionHandler
  from colossalai.auto_parallel.tensor_shard.options import ShardOption
  from colossalai.auto_parallel.tensor_shard.sharding_strategy import StrategiesVector
  from colossalai.device.device_mesh import DeviceMesh
  from colossalai.testing import clear_cache_before_run, run_on_environment_flag
label_desc: Use image node_id 2 for calling a global function with example usage: test_shard_option() without return types