def METHOD_NAME(mnist_visiondata_train, mnist_visiondata_test):
    # Act
    context = Context(train=mnist_visiondata_train, test=mnist_visiondata_test)
    # Assert
    assert_that(context, has_properties({'train': instance_of(VisionData), 'test': instance_of(VisionData)}))