def test_remove_labels_invalid_key_failure(): fs = FeatureSet("my-feature-set") fs.set_label("k1", "v1") with pytest.raises(KeyError): fs.remove_label("key1")
def test_remove_labels_empty_failure(): fs = FeatureSet("my-feature-set") with pytest.raises(KeyError): fs.remove_label("key1")