def test_get_element_by_class(self): html = ''' <span class="foo bar">nice</span> ''' self.assertEqual(get_element_by_class('foo', html), 'nice') self.assertEqual(get_element_by_class('no-such-class', html), None)
def test_get_element_by_class(self): html = """ <span class="foo bar">nice</span> """ self.assertEqual(get_element_by_class("foo", html), "nice") self.assertEqual(get_element_by_class("no-such-class", html), None)