def test_set_one_style(self):
element = Element("div")
element.set_style("background-color", "rgb(255, 0, 0)")
assert_equal(
[b'<div style="background-color: rgb(255, 0, 0)">', b"</div>"],
list(iter(element)),
)
def test_set_multiple_styles(self):
element = Element("div")
element.set_style("color", "black")
element.set_style("background-color", "rgb(255, 0, 0)")
element.set_style("display", "block")
matches = re.search(r'style="(.*)"', str(element))
css_classes = matches.group(1).split("; ")
css_classes.sort()
assert_equal(["background-color: rgb(255, 0, 0)",
"color: black",
"display: block"], css_classes)
def test_set_one_style(self):
element = Element("div")
element.set_style("background-color", "rgb(255, 0, 0)")
assert_equal([b'<div style="background-color: rgb(255, 0, 0)">',
b'</div>'], list(iter(element)))
