def testAssertionForOnlyOnePositionalInputParameter(self):
cases = [
(['a', 'b'], {}),
(['a'], {
'key': 'value'
}),
]
for args, kwargs in cases:
with self.assertRaises(AssertionError):
pipelines._ConvertInputObjectToPipelineParameters(
str, args, kwargs)
def testNoConvertionIfPipelineParameterNotStructuredObject(self):
args, kwargs = pipelines._ConvertInputObjectToPipelineParameters(
int, [1], {})
self.assertListEqual([1], args)
self.assertDictEqual({}, kwargs)
def testAssertionForInputTypeNotMatch(self):
with self.assertRaises(AssertionError):
pipelines._ConvertInputObjectToPipelineParameters(int, ['a'], {})
def testInputObjectConvertedToPipelineParametersOnlyOnce(self):
args, kwargs = pipelines._ConvertInputObjectToPipelineParameters(
_SimpleInfo, [pipelines._ENCODED_PARAMETER_FLAG], {'param': 1})
self.assertListEqual([pipelines._ENCODED_PARAMETER_FLAG], args)
self.assertDictEqual({'param': 1}, kwargs)
def testNoConvertionForSinglePipelineFutureAsInputParameter(self):
future = pipelines.pipeline.PipelineFuture(['name'])
args, kwargs = pipelines._ConvertInputObjectToPipelineParameters(
int, [future], {})
self.assertListEqual([future], args)
self.assertDictEqual({}, kwargs)