use of com.typesafe.config.Config in project aerosolve by airbnb.
the class DeleteFloatFeatureFamilyTransformTest method testEmptyFeatureVector.
@Test
public void testEmptyFeatureVector() {
Config config = ConfigFactory.parseString(makeConfig());
Transform transform = TransformFactory.createTransform(config, "test_delete_float_feature_family");
FeatureVector featureVector = new FeatureVector();
transform.doTransform(featureVector);
assertTrue(featureVector.getFloatFeatures() == null);
}
use of com.typesafe.config.Config in project aerosolve by airbnb.
the class DeleteFloatFeatureTransformTest method testEmptyFeatureVector.
@Test
public void testEmptyFeatureVector() {
Config config = ConfigFactory.parseString(makeConfig());
Transform transform = TransformFactory.createTransform(config, "test_delete");
FeatureVector featureVector = new FeatureVector();
transform.doTransform(featureVector);
assertTrue(featureVector.getStringFeatures() == null);
}
use of com.typesafe.config.Config in project aerosolve by airbnb.
the class DeleteFloatFeatureTransformTest method testTransform.
@Test
public void testTransform() {
Config config = ConfigFactory.parseString(makeConfig());
Transform transform = TransformFactory.createTransform(config, "test_delete");
FeatureVector featureVector = makeFeatureVector();
transform.doTransform(featureVector);
Map<String, Set<String>> stringFeatures = featureVector.getStringFeatures();
assertTrue(stringFeatures.size() == 1);
Map<String, Double> feat1 = featureVector.getFloatFeatures().get("loc");
assertEquals(feat1.get("lat"), 37.7, 0.1);
assertTrue(!feat1.containsKey("long"));
}
use of com.typesafe.config.Config in project aerosolve by airbnb.
the class DeleteStringFeatureFamilyTransformTest method testTransform.
@Test
public void testTransform() {
Config config = ConfigFactory.parseString(makeConfig());
Transform transform = TransformFactory.createTransform(config, "test_delete_string_feature_family");
FeatureVector featureVector = makeFeatureVector();
Map<String, Set<String>> stringFeatures = featureVector.getStringFeatures();
assertNotNull(stringFeatures);
assertTrue(stringFeatures.containsKey("strFeature1"));
assertTrue(stringFeatures.containsKey("strFeature2"));
assertTrue(stringFeatures.containsKey("strFeature3"));
assertTrue(stringFeatures.containsKey("strFeature4"));
assertEquals(4, stringFeatures.size());
transform.doTransform(featureVector);
assertNotNull(stringFeatures);
assertFalse(stringFeatures.containsKey("strFeature1"));
assertFalse(stringFeatures.containsKey("strFeature2"));
assertFalse(stringFeatures.containsKey("strFeature3"));
assertTrue(stringFeatures.containsKey("strFeature4"));
assertEquals(1, stringFeatures.size());
}
use of com.typesafe.config.Config in project aerosolve by airbnb.
the class DeleteStringFeatureTransformTest method testEmptyFeatureVector.
@Test
public void testEmptyFeatureVector() {
Config config = ConfigFactory.parseString(makeConfig());
Transform transform = TransformFactory.createTransform(config, "test_delete");
FeatureVector featureVector = new FeatureVector();
transform.doTransform(featureVector);
assertTrue(featureVector.getStringFeatures() == null);
}
Aggregations