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")