Search in sources :

Example 16 with Config

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);
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 17 with Config

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);
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 18 with Config

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"));
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 19 with Config

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());
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Set(java.util.Set) HashSet(java.util.HashSet) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 20 with Config

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);
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Aggregations

Config (com.typesafe.config.Config)138 Test (org.junit.Test)120 FeatureVector (com.airbnb.aerosolve.core.FeatureVector)117 Set (java.util.Set)35 HashSet (java.util.HashSet)29 HashMap (java.util.HashMap)19 Map (java.util.Map)19 ActorSystem (akka.actor.ActorSystem)2 ConfigObject (com.typesafe.config.ConfigObject)2 ConfigValue (com.typesafe.config.ConfigValue)2 IOException (java.io.IOException)2 URI (java.net.URI)2 ArrayList (java.util.ArrayList)2 List (java.util.List)2 DrillConfig (org.apache.drill.common.config.DrillConfig)2 BeforeClass (org.junit.BeforeClass)2 Example (com.airbnb.aerosolve.core.Example)1 Stopwatch (com.google.common.base.Stopwatch)1 ServerConfig (com.nike.riposte.server.config.ServerConfig)1 ContainerConfig (com.spotify.docker.client.messages.ContainerConfig)1