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)
