def testFeatureSetsSearch(self): path = '/featuresets/search' for dataset in self.dataRepo.getDatasets(): featureSets = dataset.getFeatureSets() request = protocol.SearchFeatureSetsRequest() request.datasetId = dataset.getId() self.verifySearchMethod(request, path, protocol.SearchFeatureSetsResponse, featureSets, self.verifyFeatureSetsEqual) for badId in self.getBadIds(): request = protocol.SearchFeatureSetsRequest() request.datasetId = badId self.verifySearchMethodFails(request, path)
def testSearchFeatureSets(self): request = protocol.SearchFeatureSetsRequest() request.datasetId = self.datasetId request.pageSize = self.pageSize self.httpClient.searchFeatureSets(self.datasetId) self.httpClient._runSearchRequest.assert_called_once_with( request, "featuresets", protocol.SearchFeatureSetsResponse)
def getAllFeatureSets(self): datasetId = self.getAllDatasets()[0].id path = 'featuresets/search' request = protocol.SearchFeatureSetsRequest() request.dataset_id = datasetId responseData = self.sendSearchRequest( path, request, protocol.SearchFeatureSetsResponse) return responseData.feature_sets
def getAllFeatureSets(self): """ Gets all feature sets available """ datasetId = self.getAllDatasets()[0].id datasetName = self.getAllDatasets()[0].name path = 'featuresets/search' request = protocol.SearchFeatureSetsRequest() request.dataset_id = datasetId responseData = self.sendSearchRequest( path, request, protocol.SearchFeatureSetsResponse) return datasetName, responseData.feature_sets
def searchFeatureSets(self, datasetId): """ Returns an iterator over the FeatureSets fulfilling the specified conditions from the specified Dataset. :param str datasetId: The ID of the :class:`ga4gh.protocol.Dataset` of interest. :return: An iterator over the :class:`ga4gh.protocol.FeatureSet` objects defined by the query parameters. """ request = protocol.SearchFeatureSetsRequest() request.dataset_id = datasetId request.page_size = pb.int(self._pageSize) return self._runSearchRequest( request, "featuresets", protocol.SearchFeatureSetsResponse)