def test_display_tasks_can_be_added(self):
v = Variant("variant name")
v.display_task(DisplayTaskDefinition("display task 0"))\
.display_tasks([DisplayTaskDefinition("display task 1"),
DisplayTaskDefinition("display task 2")])
obj = v.to_map()
assert "display task 0" == obj["display_tasks"][0]["name"]
assert "display task 1" == obj["display_tasks"][1]["name"]
assert "display task 2" == obj["display_tasks"][2]["name"]
def test_invalid_display_task(self):
v = Variant("variant name")
with pytest.raises(TypeError):
v.display_task("hello world")