def METHOD_NAME(self):
    """Test plotting using number_to_keep"""
    dist = {"00": 3, "01": 5, "11": 8, "10": 11}
    fig = plot_histogram(dist, number_to_keep=2)
    self.assertIsInstance(fig, mpl.figure.Figure)