def test_setup_pipeline_fail(self):
with self.assertRaises(TypeError):
setup.setup_pipeline(None)
def test_setup_pipeline_fail(self):
with self.assertRaises(TypeError):
setup.setup_pipeline(None)
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)
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)
