def METHOD_NAME():
    def __reader__():
        for data in test_dataset:
            image, label = data
            image = np.expand_dims(image, axis=0)
            label = np.expand_dims(label, axis=0)
            yield image, label
    return __reader__