diff --git a/tests/test_diceware.py b/tests/test_diceware.py index 2f8cc3a301e470b3990e914b5bde876f5edecfa9..0b1b881d31471715feb026a3167b8fa8da226e23 100644 --- a/tests/test_diceware.py +++ b/tests/test_diceware.py @@ -150,12 +150,15 @@ class TestHandleOptions(object): def test_handle_options_allows_plugins_updating(self, monkeypatch): # we allow plugins to update our argparser, before being used import diceware + class FakePlugin(object): @classmethod def update_argparser(cls, parser): parser.add_argument('--foo', default=2, type=int) return parser - monkeypatch.setattr(diceware, 'get_random_sources', lambda: dict(foo=FakePlugin)) + + monkeypatch.setattr( + diceware, 'get_random_sources', lambda: dict(foo=FakePlugin)) options = handle_options([]) assert options.foo == 2