def test_variants_can_be_added(self):
cd = CommandDefinition()
cd.variant("variant 0").variants(["variant 1", "variant 2"])
obj = cd.to_map()
assert "variant 0" in obj["variants"]
assert "variant 1" in obj["variants"]
assert "variant 2" in obj["variants"]
def test_invalid_variant(self):
cd = CommandDefinition()
with pytest.raises(TypeError):
cd.variant(42)
