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)