Python setup_pipelineの例

コード例 #1

ファイル: test_pipeline.py プロジェクト: mardom/corral

 def test_setup_pipeline_fail(self):
     with self.assertRaises(TypeError):
         setup.setup_pipeline(None)

コード例 #2

ファイル: test_pipeline.py プロジェクト: EdwardBetts/corral

 def test_setup_pipeline_fail(self):
     with self.assertRaises(TypeError):
         setup.setup_pipeline(None)

コード例 #3

ファイル: test_pipeline.py プロジェクト: mardom/corral

 def test_setup_pipeline(self):
     with mock.patch("atexit.register") as register:
         with mock.patch.object(TestPipeline, "setup") as pipeline_setup:
             setup.setup_pipeline(TestPipeline)
             self.assertTrue(pipeline_setup.called)
             register.assert_called_with(TestPipeline().teardown)

コード例 #4

ファイル: test_pipeline.py プロジェクト: EdwardBetts/corral

 def test_setup_pipeline(self):
     with mock.patch("atexit.register") as register:
         with mock.patch.object(TestPipeline, "setup") as pipeline_setup:
             setup.setup_pipeline(TestPipeline)
             self.assertTrue(pipeline_setup.called)
             register.assert_called_with(TestPipeline().teardown)