def METHOD_NAME(self, dataset: SeasonalContrastS2) -> None:
    x = dataset[0]
    dataset.plot(x, suptitle="Test")
    plt.close()
    dataset.plot(x, show_titles=False)
    plt.close()
    with pytest.raises(ValueError, match="doesn't support plotting"):
        x["prediction"] = torch.tensor(1)
        dataset.plot(x)