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)
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)