def test_stag(self): doc = SimpleDoc() doc.stag('img', src='/salmon-plays-piano.jpg') self.assertEqual(doc.getvalue(), '<img src="/salmon-plays-piano.jpg" />') doc = SimpleDoc(stag_end='>') doc.stag('img', src='/salmon-plays-piano.jpg') self.assertEqual(doc.getvalue(), '<img src="/salmon-plays-piano.jpg">')
def test_stag(self): doc = SimpleDoc() doc.stag('img', src = '/salmon-plays-piano.jpg') self.assertEqual( doc.getvalue(), '<img src="/salmon-plays-piano.jpg" />' ) doc = SimpleDoc(stag_end = '>') doc.stag('img', src = '/salmon-plays-piano.jpg') self.assertEqual( doc.getvalue(), '<img src="/salmon-plays-piano.jpg">' )
def test_attrs(self): doc, tag, text = SimpleDoc().tagtext() with tag('div', id='article'): if True: doc.attr(klass='new') else: doc.attr(klass='old') with tag('a', ('data-my-id', '89'), klass='alert'): text('hi') doc.stag('img', src='squirrel.jpg', klass='animal') root = ET.fromstring(doc.getvalue()) self.assertEqual(root.attrib['class'], "new") self.assertEqual(root[0].attrib['class'], "alert") self.assertEqual(root[0].attrib['data-my-id'], '89') self.assertEqual(root[1].attrib['src'], 'squirrel.jpg') self.assertEqual(root[1].attrib['class'], 'animal') self.assertRaises(KeyError, lambda: root[1].attrib['klass'])
def test_attrs(self): doc, tag, text = SimpleDoc().tagtext() with tag('div', id = 'article'): if True: doc.attr(klass = 'new') else: doc.attr(klass = 'old') with tag('a', ('data-my-id', '89'), klass='alert'): text('hi') doc.stag('img', src='squirrel.jpg', klass='animal') root = ET.fromstring(doc.getvalue()) self.assertEqual(root.attrib['class'], "new") self.assertEqual(root[0].attrib['class'], "alert") self.assertEqual(root[0].attrib['data-my-id'], '89') self.assertEqual(root[1].attrib['src'], 'squirrel.jpg') self.assertEqual(root[1].attrib['class'], 'animal') self.assertRaises( KeyError, lambda: root[1].attrib['klass'] )
def test_attributes_substitution(self): doc = SimpleDoc() doc.stag('rect', stroke_width=1) add_svg_attributes() doc.stag('circle', stroke_width=2) reset_attr_substitutions() doc.stag('line', stroke_width=3) self.assertEqual( doc.getvalue(), '<rect stroke_width="1" /><circle stroke-width="2" /><line stroke_width="3" />' )
def link(doc: SimpleDoc, path: str): """A link tag""" doc.stag("link", rel="stylesheet", type="text/css", href=path)