def test_fit_transform_poof_and_draw_calls(self): """ Test calling fit, transform, and poof on the pipeline """ pipeline = VisualPipeline([ ('a', mock.MagicMock(MockTransformer())), ('b', mock.MagicMock(MockVisualTransformer())), ('c', mock.MagicMock(MockTransformer())), ('d', mock.MagicMock(MockVisualTransformer())), ('e', mock.MagicMock(MockEstimator()),) ]) X = [[1, 1, 1, 1, 1], [2, 2, 2, 2, 2], [3, 3, 3, 3, 3]] y = [1, 2, 3, 4, 5] pipeline.fit(X, y) for name, step in pipeline.named_steps.items(): step.fit.assert_called_once_with(X, y) pipeline.transform(X) for name, step in pipeline.named_steps.items(): if name == 'e': continue step.transform.assert_called_once_with(X) pipeline.poof() for name, step in pipeline.named_steps.items(): if name in {'a', 'c', 'e'}: continue step.poof.assert_called_once_with(outpath=None)
def test_fit_transform_show_and_draw_calls(self): """ Test calling fit, transform, and show on the pipeline """ pipeline = VisualPipeline([ ("a", mock.MagicMock(MockTransformer())), ("b", mock.MagicMock(MockVisualTransformer())), ("c", mock.MagicMock(MockTransformer())), ("d", mock.MagicMock(MockVisualTransformer())), ("e", mock.MagicMock(MockEstimator())), ]) X = [[1, 1, 1, 1, 1], [2, 2, 2, 2, 2], [3, 3, 3, 3, 3]] y = [1, 2, 3, 4, 5] pipeline.fit(X, y) for name, step in pipeline.named_steps.items(): step.fit.assert_called_once_with(X, y) pipeline.transform(X) for name, step in pipeline.named_steps.items(): if name == "e": continue step.transform.assert_called_once_with(X) pipeline.show() for name, step in pipeline.named_steps.items(): if name in {"a", "c", "e"}: continue step.show.assert_called_once_with(outpath=None)