code: def test_get_best_k(self):
      self.assertEqual(trial.get_best_k(_create_trials()).id, 96)
      self.assertEqual(
          trial.get_best_k(
              _create_trials(), status_whitelist=["COMPLETED"]
          ).id,
          95,
      )
      self.assertEqual(
          trial.get_best_k(
              _create_trials(), optimization_goal="maximize"
          ).id,
          99,
      )
      self.assertLen(trial.get_best_k(_create_trials(), k=2), 2)
  
imports: from absl.testing import absltest
  from model_search.metadata import trial
label_desc: Use image node_id 1 for calling the TrialTest obj's underlying member method code with example usage: obj.test_get_best_k() without return types