mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-15 11:22:18 +08:00
BUG: Fail when passing two Filters to set_screen.
This commit is contained in:
@@ -119,3 +119,12 @@ class PipelineTestCase(TestCase):
|
||||
|
||||
p.set_screen(g, overwrite=True)
|
||||
self.assertEqual(p.screen, g)
|
||||
|
||||
with self.assertRaises(TypeError) as e:
|
||||
p.set_screen(f, g)
|
||||
|
||||
message = e.exception.args[0]
|
||||
self.assertIn(
|
||||
"expected a value of type bool or int for argument 'overwrite'",
|
||||
message,
|
||||
)
|
||||
|
||||
@@ -103,7 +103,7 @@ class Pipeline(object):
|
||||
"""
|
||||
return self.columns.pop(name)
|
||||
|
||||
@expect_types(screen=Filter)
|
||||
@expect_types(screen=Filter, overwrite=(bool, int))
|
||||
def set_screen(self, screen, overwrite=False):
|
||||
"""
|
||||
Set a screen on this Pipeline.
|
||||
|
||||
Reference in New Issue
Block a user