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')