fix use of LoadFromOPSFile

Author: dwhswenson

paths_cli/tests/wizard/test_load_from_ops.py

@@ -45,12 +45,11 @@ def ops_file_fixture():
     return storage
 
 def test_named_objs_helper(ops_file_fixture):
-    wizard = mock_wizard([])
     helper_func = named_objs_helper(ops_file_fixture, 'foo')
-    helper_func(wizard, 'any')
-    assert "what I found" in wizard.console.log_text
-    assert "bar" in wizard.console.log_text
-    assert "baz" in wizard.console.log_text
+    result = helper_func('any')
+    assert "what I found" in result
+    assert "bar" in result
+    assert "baz" in result
 
 @pytest.mark.parametrize('with_failure', [False, True])
 def test_get_ops_storage(tmpdir, with_failure):

paths_cli/wizard/load_from_ops.py

@@ -1,19 +1,22 @@
 from paths_cli.parameters import INPUT_FILE
 from paths_cli.wizard.core import get_object
 from paths_cli.wizard.standard_categories import CATEGORIES
+from paths_cli.wizard.helper import Helper
+from functools import partial
 
 from paths_cli.wizard.errors import FILE_LOADING_ERROR_MSG
 LABEL = "Load existing from OPS file"
 
 def named_objs_helper(storage, store_name):
-    def list_items(wizard, user_input):
+    def list_items(user_input, context=None):
         store = getattr(storage, store_name)
         names = [obj for obj in store if obj.is_named]
         outstr = "\n".join(['* ' + obj.name for obj in names])
-        wizard.say(f"Here's what I found:\n\n{outstr}")
+        return f"Here's what I found:\n\n{outstr}"
 
     return list_items
 
+
 @get_object
 def _get_ops_storage(wizard):
     filename = wizard.ask("What file can it be found in?",

paths_cli/wizard/wizard.py

@@ -48,10 +48,12 @@ def __init__(self, steps):
     def _patch(self):  # no-cov
         import openpathsampling as paths
         from openpathsampling.experimental.storage import monkey_patch_all
+        from paths_cli.param_core import StorageLoader
         if not self._patched:
             paths = monkey_patch_all(paths)
             paths.InterfaceSet.simstore = True
             self._patched = True
+            StorageLoader.has_simstore_patch = True
 
     def debug(content):  # no-cov
         # debug does no pretty-printing