def METHOD_NAME(self):
    import matplotlib.pyplot as plt
    cm = torch.tensor([[2, 0, 1], [0, 1, 1], [0, 0, 1]])
    fig = ConfusionMatrix.plot(cm, include_values=True, show=False)
    self.assertIsInstance(fig, plt.Figure)