def test_batch_find_return_type(self): model = TestObj() model.__doc__ = """ :param node_fill_type: How tasks should be distributed across compute nodes. Possible values include: 'spread', 'pack' :type node_fill_type: str or ~azure.batch.models.ComputeNodeFillType """ self.assertIsNone(_command_type.find_return_type(model)) model.__doc__ = """ :type callback: Callable[Bytes, response=None] :param operation_config: :ref:`Operation configuration overrides<msrest:optionsforoperations>`. :rtype: Generator or ~msrest.pipeline.ClientRawResponse """ self.assertEqual(_command_type.find_return_type(model), 'Generator')