def METHOD_NAME(mnist_visiondata_train):
    # Arrange
    context = Context(train=mnist_visiondata_train)
    # Act & Assert
    assert_that(context.have_test(), is_(False))
    assert_that(context.assert_task_type(TaskType.CLASSIFICATION), is_(True))
    assert_that(calling(context.assert_task_type).with_args(TaskType.OBJECT_DETECTION),
                raises(DeepchecksNotSupportedError, 'Check is irrelevant for task of type TaskType.CLASSIFICATION'))
    assert_that(context.get_data_by_kind(DatasetKind.TRAIN), instance_of(VisionData))
    assert_that(calling(context.get_data_by_kind).with_args(DatasetKind.TEST),
                raises(DeepchecksNotSupportedError, r'Check is irrelevant for Datasets without test dataset'))