def test_batch_find_param_type(self):
model = TestObj()
model.__doc__ = """
:param name: The name of the environment variable.
:type name: str
:param value: The value of the environment variable.
"""
self.assertEqual(_command_type.find_param_type(model, 'name'), 'str')
model.__doc__ = """
:param pool_get_options: Additional parameters for the operation
:type pool_get_options: :class:`PoolGetOptions
<azure.batch.models.PoolGetOptions>`
:param dict custom_headers: headers that will be added to the request
"""
self.assertEqual(_command_type.find_param_type(model, 'pool_get_options'),
':class:`PoolGetOptions<azure.batch.models.PoolGetOptions>`')
model.__doc__ = '''
:param node_fill_type: How tasks should be distributed across compute
nodes. Possible values include: 'spread', 'pack', 'unmapped'
:type node_fill_type: str or :class:`ComputeNodeFillType
<azure.batch.models.ComputeNodeFillType>`
"""
'''
self.assertEqual(_command_type.find_param_type(model, 'node_fill_type'),
'str or :class:`ComputeNodeFillType' +
'<azure.batch.models.ComputeNodeFillType>`')
model.__doc__ = """
:param name: The name of the environment variable.
:type name:str
:raises: BatchException
"""
self.assertEqual(_command_type.find_param_type(model, 'name'), 'str')
def test_batch_find_param_type(self):
model = TestObj()
model.__doc__ = """
:param name: The name of the environment variable.
:type name: str
:param value: The value of the environment variable.
"""
self.assertEqual(_command_type.find_param_type(model, 'name'), 'str')
model.__doc__ = """
:param pool_get_options: Additional parameters for the operation
:type pool_get_options: ~azure.batch.models.PoolGetOptions
:param dict custom_headers: headers that will be added to the request
"""
self.assertEqual(_command_type.find_param_type(model, 'pool_get_options'),
'~azure.batch.models.PoolGetOptions')
model.__doc__ = '''
:param node_fill_type: How tasks should be distributed across compute
nodes. Possible values include: 'spread', 'pack', 'unmapped'
:type node_fill_type: str or ~azure.batch.models.ComputeNodeFillType
"""
'''
self.assertEqual(_command_type.find_param_type(model, 'node_fill_type'),
'str or ~azure.batch.models.ComputeNodeFillType')
model.__doc__ = """
:param name: The name of the environment variable.
:type name:str
:raises: BatchException
"""
self.assertEqual(_command_type.find_param_type(model, 'name'), 'str')
