code: def test_where_trial_id(session: Session) -> None:
      trial = TestTrialHeartbeatModel._create_model(session)
      trial_heartbeat = TrialHeartbeatModel.where_trial_id(
          trial.trial_id, session
      )
      assert trial_heartbeat is not None
      assert isinstance(trial_heartbeat.heartbeat, datetime)
  
imports: from datetime import datetime
  import pytest
  from sqlalchemy import create_engine
  from sqlalchemy.exc import IntegrityError
  from sqlalchemy.orm import Session
  from optuna.storages._rdb.models import BaseModel
  from optuna.storages._rdb.models import StudyDirectionModel
  from optuna.storages._rdb.models import StudyModel
  from optuna.storages._rdb.models import StudySystemAttributeModel
  from optuna.storages._rdb.models import TrialHeartbeatModel
  from optuna.storages._rdb.models import TrialIntermediateValueModel
  from optuna.storages._rdb.models import TrialModel
  from optuna.storages._rdb.models import TrialSystemAttributeModel
  from optuna.storages._rdb.models import TrialUserAttributeModel
  from optuna.storages._rdb.models import TrialValueModel
  from optuna.storages._rdb.models import VersionInfoModel
  from optuna.study._study_direction import StudyDirection
  from optuna.trial import TrialState
label_desc: Use image node_id 2 for calling the TestTrialHeartbeatModel obj's underlying member method code with example usage: obj.test_where_trial_id(session) without return types