use of io.hops.hopsworks.common.featurestore.online.OnlineFeaturestoreController in project hopsworks by logicalclocks.
the class TestQueryController method setup.
@Before
public void setup() {
fs = new Featurestore();
fs.setHiveDbId(1l);
fs.setProject(new Project("test_proj"));
cachedFeaturegroup = new CachedFeaturegroup();
cachedFeaturegroup.setTimeTravelFormat(TimeTravelFormat.NONE);
fg1 = new Featuregroup(1);
fg1.setName("fg1");
fg1.setVersion(1);
fg1.setCachedFeaturegroup(cachedFeaturegroup);
fg1.setFeaturestore(fs);
fg2 = new Featuregroup(2);
fg2.setName("fg2");
fg2.setVersion(1);
fg2.setCachedFeaturegroup(cachedFeaturegroup);
fg2.setFeaturestore(fs);
fg3 = new Featuregroup(3);
fg3.setName("fg3");
fg3.setVersion(1);
fg3.setCachedFeaturegroup(cachedFeaturegroup);
fg3.setFeaturestore(fs);
fg4 = new Featuregroup(4);
fg4.setName("fg4");
fg4.setVersion(1);
fg4.setCachedFeaturegroup(cachedFeaturegroup);
fg4.setFeaturestore(fs);
fgHudi = new Featuregroup(5);
fgHudi.setName("fgHudi");
fgHudi.setVersion(1);
hudiFeatureGroup = new CachedFeaturegroup();
hudiFeatureGroup.setTimeTravelFormat(TimeTravelFormat.HUDI);
fgHudi.setCachedFeaturegroup(hudiFeatureGroup);
fgHudi.setFeaturestore(fs);
fg1Features = new ArrayList<>();
fg1Features.add(new Feature("pr", "", true));
fg1Features.add(new Feature("fg1_ft2", "", false));
fg1FeaturesDTO = new ArrayList<>();
fg1FeaturesDTO.add(new FeatureGroupFeatureDTO("pr", "Integer", "", true, false, "", null));
fg1FeaturesDTO.add(new FeatureGroupFeatureDTO("fg1_ft2", "String", "", false, false, "", null));
fg2Features = new ArrayList<>();
fg2Features.add(new Feature("pr", "", true));
fg2Features.add(new Feature("fg2_ft2", "", false));
fg2FeaturesDTO = new ArrayList<>();
fg2FeaturesDTO.add(new FeatureGroupFeatureDTO("pr", "Integer", "", true, false, "", null));
fg2FeaturesDTO.add(new FeatureGroupFeatureDTO("fg2_ft2", "String", "", false, false, "", null));
fg3Features = new ArrayList<>();
fg3Features.add(new Feature("fg3_ft1", "", true));
fg3Features.add(new Feature("fg3_ft2", "", false));
fg4Features = new ArrayList<>();
fg4Features.add(new Feature("pr", "fg4", true));
fg4Features.add(new Feature("fg4_ft4_1", "fg4", "Float", null, "prefix4_"));
fg4Features.add(new Feature("fg4_ft4_2", "fg4", "Float", null, "prefix4_"));
fg4Features.add(new Feature("_hoodie_record_key", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_partition_path", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_commit_time", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_file_name", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_commit_seqno", "fg4", "String", null, null));
singleEqualsJoinOperator = Arrays.asList(SqlCondition.EQUALS);
featuregroupController = Mockito.mock(FeaturegroupController.class);
featuregroupFacade = Mockito.mock(FeaturegroupFacade.class);
featurestoreFacade = Mockito.mock(FeaturestoreFacade.class);
onlineFeaturestoreController = Mockito.mock(OnlineFeaturestoreController.class);
cachedFeaturegroupController = Mockito.mock(CachedFeaturegroupController.class);
project = Mockito.mock(Project.class);
user = Mockito.mock(Users.class);
filterController = new FilterController(new ConstructorController());
target = new QueryController(featuregroupController, featuregroupFacade, filterController, featurestoreFacade, onlineFeaturestoreController);
new JoinController(new ConstructorController());
}
use of io.hops.hopsworks.common.featurestore.online.OnlineFeaturestoreController in project hopsworks by logicalclocks.
the class TestConstructorController method setup.
@Before
public void setup() {
fs = new Featurestore();
fs.setHiveDbId(1l);
fs.setProject(new Project("test_proj"));
cachedFeaturegroup = new CachedFeaturegroup();
cachedFeaturegroup.setTimeTravelFormat(TimeTravelFormat.NONE);
fg1 = new Featuregroup(1);
fg1.setName("fg1");
fg1.setVersion(1);
fg1.setCachedFeaturegroup(cachedFeaturegroup);
fg1.setFeaturestore(fs);
fg2 = new Featuregroup(2);
fg2.setName("fg2");
fg2.setVersion(1);
fg2.setCachedFeaturegroup(cachedFeaturegroup);
fg2.setFeaturestore(fs);
fg3 = new Featuregroup(3);
fg3.setName("fg3");
fg3.setVersion(1);
fg3.setCachedFeaturegroup(cachedFeaturegroup);
fg3.setFeaturestore(fs);
fg4 = new Featuregroup(4);
fg4.setName("fg4");
fg4.setVersion(1);
fg4.setCachedFeaturegroup(cachedFeaturegroup);
fg4.setFeaturestore(fs);
fgHudi = new Featuregroup(5);
fgHudi.setName("fgHudi");
fgHudi.setVersion(1);
hudiFeatureGroup = new CachedFeaturegroup();
hudiFeatureGroup.setTimeTravelFormat(TimeTravelFormat.HUDI);
fgHudi.setCachedFeaturegroup(hudiFeatureGroup);
fgHudi.setFeaturestore(fs);
fg1Features = new ArrayList<>();
fg1Features.add(new Feature("pr", "", true));
fg1Features.add(new Feature("fg1_ft2", "", false));
fg1FeaturesDTO = new ArrayList<>();
fg1FeaturesDTO.add(new FeatureGroupFeatureDTO("pr", "Integer", "", true, false, "", null));
fg1FeaturesDTO.add(new FeatureGroupFeatureDTO("fg1_ft2", "String", "", false, false, "", null));
fg2Features = new ArrayList<>();
fg2Features.add(new Feature("pr", "", true));
fg2Features.add(new Feature("fg2_ft2", "", false));
fg2FeaturesDTO = new ArrayList<>();
fg2FeaturesDTO.add(new FeatureGroupFeatureDTO("pr", "Integer", "", true, false, "", null));
fg2FeaturesDTO.add(new FeatureGroupFeatureDTO("fg2_ft2", "String", "", false, false, "", null));
fg3Features = new ArrayList<>();
fg3Features.add(new Feature("fg3_ft1", "", true));
fg3Features.add(new Feature("fg3_ft2", "", false));
fg4Features = new ArrayList<>();
fg4Features.add(new Feature("pr", "fg4", true));
fg4Features.add(new Feature("fg4_ft4_1", "fg4", "Float", null, "prefix4_"));
fg4Features.add(new Feature("fg4_ft4_2", "fg4", "Float", null, "prefix4_"));
fg4Features.add(new Feature("_hoodie_record_key", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_partition_path", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_commit_time", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_file_name", "fg4", "String", null, null));
fg4Features.add(new Feature("_hoodie_commit_seqno", "fg4", "String", null, null));
singleEqualsJoinOperator = Arrays.asList(SqlCondition.EQUALS);
featuregroupController = Mockito.mock(FeaturegroupController.class);
featuregroupFacade = Mockito.mock(FeaturegroupFacade.class);
featurestoreFacade = Mockito.mock(FeaturestoreFacade.class);
onlineFeaturestoreController = Mockito.mock(OnlineFeaturestoreController.class);
cachedFeaturegroupController = Mockito.mock(CachedFeaturegroupController.class);
project = Mockito.mock(Project.class);
user = Mockito.mock(Users.class);
filterController = new FilterController(new ConstructorController());
target = new ConstructorController(featuregroupController, featurestoreFacade, featuregroupFacade, onlineFeaturestoreController, cachedFeaturegroupController, filterController, new JoinController(new ConstructorController()));
new JoinController(new ConstructorController());
queryController = new QueryController(featuregroupController, featuregroupFacade, filterController, featurestoreFacade, onlineFeaturestoreController);
}
use of io.hops.hopsworks.common.featurestore.online.OnlineFeaturestoreController in project hopsworks by logicalclocks.
the class TestPitJoinController method setup.
@Before
public void setup() {
fs = new Featurestore();
fs.setHiveDbId(1l);
fs.setProject(new Project("test_proj"));
cachedFeaturegroup = new CachedFeaturegroup();
cachedFeaturegroup.setTimeTravelFormat(TimeTravelFormat.NONE);
fgLeft = new Featuregroup(1);
fgLeft.setEventTime("ts");
fgLeft.setName("fg0");
fgLeft.setVersion(1);
fgLeft.setCachedFeaturegroup(cachedFeaturegroup);
fgLeft.setFeaturestore(fs);
fgRight = new Featuregroup(2);
fgRight.setEventTime("ts");
fgRight.setName("fg1");
fgRight.setVersion(1);
fgRight.setCachedFeaturegroup(cachedFeaturegroup);
fgRight.setFeaturestore(fs);
fgRight1 = new Featuregroup(3);
fgRight1.setEventTime("ts");
fgRight1.setName("fg2");
fgRight1.setVersion(1);
fgRight1.setCachedFeaturegroup(cachedFeaturegroup);
fgRight1.setFeaturestore(fs);
FeaturegroupController featuregroupController = Mockito.mock(FeaturegroupController.class);
FeaturegroupFacade featuregroupFacade = Mockito.mock(FeaturegroupFacade.class);
FeaturestoreFacade featurestoreFacade = Mockito.mock(FeaturestoreFacade.class);
OnlineFeaturestoreController onlineFeaturestoreController = Mockito.mock(OnlineFeaturestoreController.class);
CachedFeaturegroupController cachedFeaturegroupController = Mockito.mock(CachedFeaturegroupController.class);
FilterController filterController = new FilterController(new ConstructorController());
JoinController joinController = new JoinController(new ConstructorController());
ConstructorController constructorController = new ConstructorController(featuregroupController, featurestoreFacade, featuregroupFacade, onlineFeaturestoreController, cachedFeaturegroupController, filterController, joinController);
pitJoinController = new PitJoinController(constructorController, filterController, joinController);
}
Aggregations