def testReadFromExamplesArtifact_UptoMaxExamples(self):
builder = request_builder.RequestBuilder(max_examples=1,
model_name='foo')
builder.ReadFromExamplesArtifact(self._examples)
self.assertEqual(builder.num_examples, 1)
def testReadFromExamplesArtifact_NoOpIfExamplesFulfilled(self):
builder = request_builder.RequestBuilder(max_examples=1,
model_name='foo')
builder.ReadFromExamplesArtifact(self._examples)
with mock.patch.object(builder, '_ReadFromDataset') as read_impl:
builder.ReadFromExamplesArtifact(self._examples)
read_impl.assert_not_called()
def testReadFromExamplesArtifact_FailsIfFileNotExists(self):
invalid_uri_examples = standard_artifacts.Examples()
invalid_uri_examples.split_names = self._examples.split_names
invalid_uri_examples.uri = '/not/existing/path'
builder = request_builder.RequestBuilder(max_examples=1,
model_name='foo')
with self.assertRaisesRegexp(ValueError, 'Unable to find examples'):
builder.ReadFromExamplesArtifact(invalid_uri_examples)
def testReadFromExamplesArtifact_FailsIfNoSplitNames(self):
examples = standard_artifacts.Examples()
examples.split_names = artifact_utils.encode_split_names([])
examples.uri = self._examples.uri
builder = request_builder.RequestBuilder(max_examples=1,
model_name='foo')
with self.assertRaisesRegexp(ValueError, 'No split_name is available'):
builder.ReadFromExamplesArtifact(examples)
def testBuildRegressionRequest(self):
builder = request_builder.RequestBuilder(max_examples=1,
model_name='foo')
builder.ReadFromExamplesArtifact(self._examples)
requests = builder.BuildRegressionRequests()
for request in requests:
self.assertIsInstance(request, regression_pb2.RegressionRequest)
self.assertEqual(request.model_spec.name, 'foo')
self.assertEqual(request.model_spec.signature_name, '')
for example in request.input.example_list.examples:
self.assertValidTaxiExample(example)
def testReadFromExamplesArtifact_FailsIfSplitNotExists(self):
builder = request_builder.RequestBuilder(max_examples=1,
model_name='foo')
with self.assertRaisesRegexp(ValueError, 'No split_name asdf'):
builder.ReadFromExamplesArtifact(self._examples, split_name='asdf')
def testInit_FailsIfMaxExamplesInvalid(self):
with self.assertRaises(ValueError):
request_builder.RequestBuilder(max_examples=-1, model_name='foo')
