コード例 #1
0
    def test_poolmgr_create_spec(self, mock_pool, mock_api, mock_creds, mock_cfg):
        """Test create_spec"""

        mgr = PoolManager(mock_creds, cfg=mock_cfg)
        spec = mgr.create_spec()
        mock_pool.assert_called_with(mock.ANY, 0, 1, False)

        spec = mgr.create_spec(target_size=3, max_tasks=3, communication=True)
        mock_pool.assert_called_with(mock.ANY, 3, 3, True)
コード例 #2
0
    def test_poolmgr_create_spec(self, mock_pool, mock_api, mock_creds,
                                 mock_cfg):
        """Test create_spec"""

        mgr = PoolManager(mock_creds, cfg=mock_cfg)
        spec = mgr.create_spec()
        mock_pool.assert_called_with(mock.ANY, 0, 1, False)

        spec = mgr.create_spec(target_size=3, max_tasks=3, communication=True)
        mock_pool.assert_called_with(mock.ANY, 3, 3, True)