Python pool_setの例

コード例 #1

ファイル: test_pool_command.py プロジェクト: yunfengzhou-hub/flink-ai-extended

    def test_pool_list(self):
        pool_command.pool_set(
            self.parser.parse_args(['pools', 'set', 'foo', '1', 'test']))
        with redirect_stdout(io.StringIO()) as stdout:
            pool_command.pool_list(self.parser.parse_args(['pools', 'list']))

        self.assertIn('foo', stdout.getvalue())

コード例 #2

    def test_pool_list(self):
        pool_command.pool_set(
            self.parser.parse_args(['pools', 'set', 'foo', '1', 'test']))
        with self.assertLogs(level='INFO') as cm:
            pool_command.pool_list(self.parser.parse_args(['pools', 'list']))

        stdout = cm.output

        self.assertIn('foo', stdout[0])

コード例 #3

ファイル: test_pool_command.py プロジェクト: ztz0223/airflow

 def test_pool_delete(self):
     pool_command.pool_set(
         self.parser.parse_args(['pools', 'set', 'foo', '1', 'test']))
     pool_command.pool_delete(
         self.parser.parse_args(['pools', 'delete', 'foo']))
     self.assertEqual(self.session.query(Pool).count(), 1)

コード例 #4

ファイル: test_pool_command.py プロジェクト: ztz0223/airflow

 def test_pool_get(self):
     pool_command.pool_set(
         self.parser.parse_args(['pools', 'set', 'foo', '1', 'test']))
     pool_command.pool_get(self.parser.parse_args(['pools', 'get', 'foo']))

コード例 #5

ファイル: test_pool_command.py プロジェクト: ChethanUK/airflow-1

 def test_pool_create(self):
     pool_command.pool_set(self.parser.parse_args(['pools', 'set', 'foo', '1', 'test']))
     assert self.session.query(Pool).count() == 2