import unittest import base class Suite(base.Base): def test_1(self): """Test case 1""" self.start() el = self.xpath('button') el.click() alert = self.driver.switch_to.alert self.assertEqual("3", alert.text)