def test_add_sample_size_to_xtick_labels_extra_classes(self): labels = _add_sample_size_to_xtick_labels(self.a, [0, 'a', 'b', 'bb', 'c']) exp = ['0 (n=0)', 'a (n=2)', 'b (n=2)', 'bb (n=0)', 'c (n=2)'] self.assertListEqual(labels, exp)
def test_add_sample_size_to_xtick_labels(self): labels = _add_sample_size_to_xtick_labels(self.a, ['a', 'b', 'c']) exp = ['a (n=2)', 'b (n=2)', 'c (n=2)'] self.assertListEqual(labels, exp)