code: def test_aug_branin(self) -> None:
      self.assertEqual(aug_branin.name, "Aug_Branin")
      self.assertAlmostEqual(aug_branin(1, 2, 1), 21.62763539206238)
      self.assertAlmostEqual(
          aug_branin(x1=1, x2=2, x3=1), 21.62763539206238
      )
      self.assertAlmostEqual(
          aug_branin(np.array([1, 2, 1])), 21.62763539206238
      )
      self.assertAlmostEqual(
          # pyre-fixme[16]: Item `float` of `Union[float, ndarray]` has no
          #  attribute `__getitem__`.
          aug_branin(np.array([[1, 2, 1], [1, 2, 1]]))[0],
          21.62763539206238,
      )
      self.assertAlmostEqual(aug_branin.minimums[0][0], -np.pi)
      self.assertAlmostEqual(aug_branin.fmin, branin.fmin)
      self.assertAlmostEqual(aug_branin.fmax, branin.fmax)
      self.assertEqual(aug_branin.domain[2], (0, 1))
      self.assertEqual(aug_branin.domain[0], (-5, 10))
      self.assertEqual(aug_branin.required_dimensionality, 3)
      with self.assertRaisesRegex(
          NotImplementedError, "Aug_Branin does not specify"
      ):
          aug_branin.maximums
      with self.assertRaisesRegex(
          ValueError, "Synthetic function call"
      ):
          aug_branin(np.array([[[1, 3, 1]]]))
  
file_path: Ax/ax/utils/measurement/tests/test_synthetic_functions.py
imports: import numpy
  from ax.utils.common.testutils import TestCase
  from ax.utils.measurement.synthetic_functions import aug_branin, aug_hartmann6, branin, FromBotorch, hartmann6
  from botorch.test_functions import synthetic
label_desc: Use image node_id 4 for calling the TestSyntheticFunctions obj's underlying member method code with example usage: obj.test_aug_branin() without return types