diff options
Diffstat (limited to 'tests/test_cli_main.py')
| -rw-r--r-- | tests/test_cli_main.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/test_cli_main.py b/tests/test_cli_main.py index 2f89268..e415284 100644 --- a/tests/test_cli_main.py +++ b/tests/test_cli_main.py @@ -165,6 +165,11 @@ def wrap_mock_command(mock: MagicMock, command: Callable) -> Callable: pruning_target=None, clustering_target=None, backend=None, + rewrite=False, + rewrite_target=None, + rewrite_start=None, + rewrite_end=None, + dataset=None, ), ], [ @@ -189,6 +194,11 @@ def wrap_mock_command(mock: MagicMock, command: Callable) -> Callable: pruning_target=0.5, clustering_target=32, backend=None, + rewrite=False, + rewrite_target=None, + rewrite_start=None, + rewrite_end=None, + dataset=None, ), ], [ @@ -210,6 +220,11 @@ def wrap_mock_command(mock: MagicMock, command: Callable) -> Callable: pruning_target=None, clustering_target=None, backend=["some_backend"], + rewrite=False, + rewrite_target=None, + rewrite_start=None, + rewrite_end=None, + dataset=None, ), ], [ |