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)
