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"