code: def test_load_dataset_by_id(self, mock_dataset, mock_get_db_conn):
      # Setup
      identifier = ObjectId()
      mock_db = MagicMock()
      mock_get_db_conn.return_value = mock_db
      mock_db.datasets.find_one.return_value = {
          "_id": ObjectId(identifier),
          "name": "test_dataset",
      }
  
      # Test
      result = load_dataset(id=identifier)
  
      # Assertions
      mock_get_db_conn.assert_called_once()
      mock_db.datasets.find_one.assert_called_once_with(
          {"_id": ObjectId(identifier)}, {"name": True}
      )
  
      self.assertEqual(result, mock_dataset.return_value)
  
imports: import time
  import unittest
  from unittest.mock import MagicMock, patch
  import numpy
  import fiftyone
  import fiftyone.constants
  import fiftyone.core.media
  import fiftyone.core.odm
  import fiftyone.core.utils
  from fiftyone.migrations.runner import MigrationRunner
  from decorators import drop_datasets
  from bson import ObjectId
  from fiftyone.core.odm.utils import load_dataset
label_desc: Use image node_id 1 for calling the TestLoadDataset obj's underlying member method code with example usage: obj.test_load_dataset_by_id(mock_dataset, mock_get_db_conn) without return types