code: def test_calculate_2_shards_with_complete_test_times(self) -> None:
      expected_shards = [
          (
              60.0,
              [
                  ShardedTest(
                      test="super_long_test",
                      shard=1,
                      num_shards=1,
                      time=55,
                  ),
                  ShardedTest(
                      test="normal_test3", shard=1, num_shards=1, time=5
                  ),
              ],
          ),
          (
              58.31,
              [
                  ShardedTest(
                      test="long_test1", shard=1, num_shards=1, time=22
                  ),
                  ShardedTest(
                      test="long_test2", shard=1, num_shards=1, time=18
                  ),
                  ShardedTest(
                      test="normal_test1", shard=1, num_shards=1, time=9
                  ),
                  ShardedTest(
                      test="normal_test2", shard=1, num_shards=1, time=7
                  ),
                  ShardedTest(
                      test="short_test1", shard=1, num_shards=1, time=1
                  ),
                  ShardedTest(
                      test="short_test2",
                      shard=1,
                      num_shards=1,
                      time=0.6,
                  ),
                  ShardedTest(
                      test="short_test3",
                      shard=1,
                      num_shards=1,
                      time=0.4,
                  ),
                  ShardedTest(
                      test="short_test4",
                      shard=1,
                      num_shards=1,
                      time=0.3,
                  ),
                  ShardedTest(
                      test="short_test5",
                      shard=1,
                      num_shards=1,
                      time=0.01,
                  ),
              ],
          ),
      ]
      self.assert_shards_equal(
          expected_shards,
          calculate_shards(
              2, self.tests, self.test_times, self.test_class_times
          ),
      )
  
file_path: pytorch/tools/test/test_test_selections.py
imports: import pathlib
  import random
  import sys
  import unittest
  from collections import defaultdict
  from typing import Dict, List, Tuple
label_desc: Use image node_id 2 for calling the TestCalculateShards obj's underlying member method code with example usage: obj.test_calculate_2_shards_with_complete_test_times() without return types