def test_invalid_modules_cannot_be_added(self): v = Variant("variant name") with pytest.raises(TypeError): v.module(25) with pytest.raises(TypeError): v.modules(TaskSpec("not array"))
def test_modules_can_be_added(self): v = Variant("variant name") v.modules(["enterprise", "enterprise2"]) obj = v.to_map() assert "enterprise" in obj["modules"] assert "enterprise2" in obj["modules"]