def test_sum(self): calculator = Calulator(self.driver) calculator.sum_numbers(3, 4) assert calculator.get_element(calculator.sum_label).text == '7'