def METHOD_NAME(predictor, test_loader, accept):
    predictor.deserializer = ACCEPT_TYPE_TO_DESERIALIZER_MAP[accept]
    data = _get_mnist_batch(test_loader).view(test_loader.batch_size, -1)
    output = predictor.predict(data)
    assert np.asarray(output).shape == (test_loader.batch_size, 10)