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