def METHOD_NAME(self):
    image_filenames = [
        os.path.join(self.test_data, 'infer_data', 'lindau_000001_000019.png'),
        os.path.join(self.test_data, 'infer_data', 'lindau_000003_000019.png'),
    ]
    images = [Image.open(f) for f in image_filenames]
    learner = EfficientPsLearner(self.config_file)
    learner.load(self.model_weights)
    predictions = learner.infer(images)
    for prediction in predictions:
        for heatmap in prediction:
            self.assertIsInstance(heatmap, Heatmap)