def METHOD_NAME(mnist_visiondata_train, mnist_visiondata_test):
    mnist_visiondata_train.name = 'Ref'
    mnist_visiondata_test.name = 'Win'
    result = HeatmapComparison().run(mnist_visiondata_train, mnist_visiondata_test)
    assert_that(result.display[0].layout.annotations[0].text, 'Ref')
    assert_that(result.display[0].layout.annotations[1].text, 'Win')