def test_pie_labels(self): pie_chart = PieChart('pie title', ['one', 'two'], [1, 2]) self.assertEqual(pie_chart.test_labels(), ['one', 'two'])