6fac2b4658a751ceed8d52bdfae1fa08af495f8c,tests/providers/gcp/gcp_tpu_test.py,GcpTpuTestCase,setUp,#GcpTpuTestCase#,52

Before Change

  def setUp(self):
    flag_values = {"run_uri": "123", "project": None, "tpu_cores_per_donut": 8}

    p = mock.patch(gcp_tpu.__name__ + ".FLAGS")
    flags_mock = p.start()
    flags_mock.configure_mock(**flag_values)
    self.addCleanup(p.stop)
    mock_tpu_spec_attrs = self.CreateTpuSpecDict()
    self.mock_tpu_spec = mock.Mock(
        spec=benchmark_config_spec._TpuGroupSpec)
    self.mock_tpu_spec.configure_mock(**mock_tpu_spec_attrs)

After Change

    return tpu_class

  def setUp(self):
    self.flags = mock_flags.PatchTestCaseFlags(self)
    self.flags.run_uri = "123"
    self.flags.project = ""
    self.flags.tpu_cores_per_donut = 8
    self.flags.gcloud_path = "gcloud"

    mock_tpu_spec_attrs = self.CreateTpuSpecDict()
    self.mock_tpu_spec = mock.Mock(
        spec=benchmark_config_spec._TpuGroupSpec)

Instances