code: def test_update_dataset(self, mock_update_dataset):
      self.hook.update_dataset(
          dataset=DATASET,
          update_mask=MASK,
      )
  
      mock_update_dataset.assert_called_once_with(
          request=dict(dataset=DATASET, update_mask=MASK),
          retry=DEFAULT,
          timeout=None,
          metadata=(),
      )
  
imports: from __future__ import annotations
  from unittest import mock
  import pytest
  from google.api_core.gapic_v1.method import DEFAULT
  from google.cloud.automl_v1beta1 import AutoMlClient
  from airflow.providers.google.cloud.hooks.automl import CloudAutoMLHook
  from airflow.providers.google.common.consts import CLIENT_INFO
  from tests.providers.google.cloud.utils.base_gcp_mock import mock_base_gcp_hook_no_default_project_id
label_desc: Use image node_id 13 for calling the TestAutoMLHook obj's underlying member method code with example usage: obj.test_update_dataset(mock_update_dataset) without return types