Fixed test

UCSD-E4E · Mar 12, 2023 · 5c8e152 · 5c8e152
1 parent e9a3d62
commit 5c8e152
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -37,11 +37,11 @@ def test_init_dataset_today(test_app: Tuple[Mock, DataManager, Path]):
     Args:
         test_app (Tuple[Mock, DataManager, Path]): Test application
     """
-    mock = test_app
+    mock, _, _ = test_app
     args = split('e4edm init_dataset --date today --project TEST --location Location')
     with patch('sys.argv', args):
         main()
-        mock.assert_called_once_with(
+        mock.initialize_dataset.assert_called_once_with(
             date=dt.date.today(),
             project='TEST',
             location='Location',