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_module_can_be_added(self):
v = Variant("variant name")
v.module("enterprise")
obj = v.to_map()
assert ["enterprise"] == obj["modules"]
