def test_make_output_with_not_resource_name_returns_serialized_list_json(
self
):
output_object = ["a", 2]
expected_result = '["a", 2]'
result = remote_runner.make_output(output_object)
self.assertEqual(result, expected_result)
def test_make_output_with_resource_name_returns_resource_name_value(self):
output_object = aiplatform.Model._empty_constructor(
project='test-project')
output_object._gca_resource = aiplatform.gapic.Model(
name='test_resource_name')
expected_result = 'test_resource_name'
result = remote_runner.make_output(output_object)
self.assertEqual(result, expected_result)