use of io.hops.hopsworks.persistence.entity.featurestore.Featurestore in project hopsworks by logicalclocks.
the class TestTransformationFunctionController method setup.
@Before
public void setup() {
project = Mockito.mock(Project.class);
user = Mockito.mock(Users.class);
fs = new Featurestore();
transformationFunctionFacade = Mockito.mock(TransformationFunctionFacade.class);
transformationFunctionController = new TransformationFunctionController(transformationFunctionFacade);
}
use of io.hops.hopsworks.persistence.entity.featurestore.Featurestore in project hopsworks by logicalclocks.
the class TestFeatureGroupCommitController method setup.
@Before
public void setup() {
Inode inode = new Inode();
HiveSds hiveSds = new HiveSds();
hiveSds.setSdId(1l);
hiveSds.setLocation("hopsfs://namenode.service.consul:8020/apps/hive/warehouse/test_proj_featurestore.db/fg1_1");
hiveSds.setInode(inode);
HiveTbls hiveTbls = new HiveTbls();
hiveTbls.setSdId(hiveSds);
hiveTbls.setTblName("fg1_1");
CachedFeaturegroup cachedFeaturegroup = new CachedFeaturegroup();
cachedFeaturegroup.setHiveTbls(hiveTbls);
fs = new Featurestore();
fs.setHiveDbId(1l);
fs.setProject(new Project("test_proj"));
fg1 = new Featuregroup(1);
fg1.setName("fg1_1");
fg1.setVersion(1);
fg1.setFeaturestore(fs);
fg1.setCachedFeaturegroup(cachedFeaturegroup);
}
use of io.hops.hopsworks.persistence.entity.featurestore.Featurestore 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.persistence.entity.featurestore.Featurestore 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);
}
use of io.hops.hopsworks.persistence.entity.featurestore.Featurestore in project hopsworks by logicalclocks.
the class TestFilterController method setup.
@Before
public void setup() {
constructorController = new ConstructorController();
filterController = new FilterController(constructorController);
Featurestore fs = new Featurestore();
fs.setHiveDbId(1l);
fs.setProject(new Project("test_proj"));
CachedFeaturegroup 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);
fgLookup = new HashMap<>();
fgLookup.put(fg1.getId(), fg1);
fgLookup.put(fg2.getId(), fg2);
fgLookup.put(fg3.getId(), fg3);
fg1Features = new ArrayList<>();
fg1Features.add(new Feature("fg1_pk", "fg1", "string", true, null, "prefix2_", fg1));
fg1Features.add(new Feature("fg1_ft", "fg1", "integer", false, null, "prefix2_", fg1));
fg1Features.add(new Feature("join", "fg1", "string", true, null, "prefix2_", fg1));
fg2Features = new ArrayList<>();
fg2Features.add(new Feature("fg2_pk", "fg2", "string", true, null, "prefix2_"));
fg2Features.add(new Feature("fg2_ft", "fg2", "double", false, "10.0", "prefix2_"));
fg3Features.add(new Feature("join", "fg2", "string", true, null, "prefix2_"));
fg3Features = new ArrayList<>();
fg3Features.add(new Feature("fg3_pk", "fg3", "string", true, null, "prefix3_"));
fg3Features.add(new Feature("fg3_ft", "fg3", "string", false, "default", "prefix3_"));
fg3Features.add(new Feature("join", "fg3", "string", true, null, "prefix3_"));
joinFeatures = new ArrayList<>();
joinFeatures.add(fg1Features.get(2));
leftOn = new ArrayList<>();
leftOn.add(fg1Features.get(0));
leftOn.add(fg1Features.get(2));
rightOn = new ArrayList<>();
rightOn.add(fg3Features.get(0));
rightOn.add(fg3Features.get(2));
availableFeatureLookup = new HashMap<>();
availableFeatureLookup.put(fg1.getId(), fg1Features);
availableFeatureLookup.put(fg2.getId(), fg2Features);
availableFeatureLookup.put(fg3.getId(), fg3Features);
}
Aggregations