def test_unequal_feature_set_based_on_labels():
fs1 = FeatureSet("my-feature-set")
fs2 = FeatureSet("my-feature-set")
assert fs1 == fs2
fs1.set_label("k1", "v1")
fs2.set_label("k1", "v1")
assert fs1 == fs2
fs2.set_label("k1", "unequal")
assert not fs1 == fs2
def test_unequal_feature_set_other_has_no_labels():
fs1 = FeatureSet("my-feature-set")
fs2 = FeatureSet("my-feature-set")
assert fs1 == fs2
fs1.set_label("k1", "v1")
assert not fs1 == fs2
def test_set_labels_overwrites_existing():
fs = FeatureSet("my-feature-set")
fs.set_label("k1", "v1")
fs.set_label("k1", "v2")
assert fs.labels["k1"] == "v2"
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_set_label_feature_set():
fs = FeatureSet("my-feature-set")
fs.set_label("k1", "v1")
assert fs.labels["k1"] == "v1"
