code: def test_create_notebook_wait_for_completion(self, mock_hook_conn):
      operator = SageMakerCreateNotebookOperator(
          task_id="task_test",
          instance_name=INSTANCE_NAME,
          instance_type=INSTANCE_TYPE,
          role_arn=ROLE_ARN,
      )
      operator.execute(None)
      mock_hook_conn.create_notebook_instance.assert_called_once()
      mock_hook_conn.get_waiter.assert_called_once_with(
          "notebook_instance_in_service"
      )
  
imports: from __future__ import annotations
  from typing import Generator
  from unittest import mock
  import pytest
  from moto import mock_sagemaker
  from airflow.providers.amazon.aws.hooks.sagemaker import SageMakerHook
  from airflow.providers.amazon.aws.operators.sagemaker import SageMakerCreateNotebookOperator, SageMakerDeleteNotebookOperator, SageMakerStartNoteBookOperator, SageMakerStopNotebookOperator
label_desc: Use image node_id 2 for calling the TestSagemakerCreateNotebookOperator obj's underlying member method code with example usage: obj.test_create_notebook_wait_for_completion(mock_hook_conn) without return types