def METHOD_NAME(predictor):
    batch_size = 100
    data = numpy.zeros((batch_size, 784), dtype="float32")
    output = predictor.predict(data)
    assert len(output) == batch_size
    data = numpy.zeros((batch_size, 1, 28, 28), dtype="float32")
    output = predictor.predict(data)
    assert len(output) == batch_size
    data = numpy.zeros((batch_size, 28, 28), dtype="float32")
    output = predictor.predict(data)
    assert len(output) == batch_size