def test_view_pair_stair_page_should_see_add_button(self): pair_stair_page = PairStairPage(self.driver) self.driver.go_to(PAIR_STAIR_PAGE) self.assertEqual(pair_stair_page.get_add_button().text, 'add')