import unittest import base class Suite(base.Base): def test_1(self): """Test case 1""" self.start('main') l = self.xpath('li[1]/a') l.click() self.assertEqual("Yes!", self.body_text()) def test_2(self): """Test case 2""" self.start('main') l = self.xpath('li[2]/a') l.click() self.assertEqual("No!", self.body_text())