def test_remove_feature_failure(self): with pytest.raises(ValueError): fs = FeatureSet("my-feature-set") fs.drop(name="my-feature-1")
def test_add_remove_features_success(self): fs = FeatureSet("my-feature-set") fs.add(Feature(name="my-feature-1", dtype=ValueType.INT64)) fs.add(Feature(name="my-feature-2", dtype=ValueType.INT64)) fs.drop(name="my-feature-1") assert len(fs.features) == 1 and fs.features[0].name == "my-feature-2"