Tests for two_state_tps.py

Author: dwhswenson

paths_cli/tests/wizard/conftest.py

@@ -68,3 +68,12 @@ def toy_engine():
         topology=topology
     ).named('toy-engine')
     return engine
+
+
+@pytest.fixture
+def tps_network():
+    cv = paths.CoordinateFunctionCV('x', lambda s: s.xyz[0][0])
+    state_A = paths.CVDefinedVolume(cv, float("-inf"), 0).named("A")
+    state_B = paths.CVDefinedVolume(cv, 0, float("inf")).named("B")
+    network = paths.TPSNetwork(state_A, state_B).named('tps-network')
+    return network

paths_cli/tests/wizard/test_tps.py

@@ -10,16 +10,6 @@
 
 from paths_cli.wizard.tps import tps_scheme
 
-
-
-@pytest.fixture
-def tps_network():
-    cv = paths.CoordinateFunctionCV('x', lambda s: s.xyz[0][0])
-    state_A = paths.CVDefinedVolume(cv, float("-inf"), 0).named("A")
-    state_B = paths.CVDefinedVolume(cv, 0, float("inf")).named("B")
-    network = paths.TPSNetwork(state_A, state_B).named('tps-network')
-    return network
-
 @pytest.mark.parametrize('as_scheme', [False, True])
 def test_tps_scheme(tps_network, toy_engine, as_scheme):
     wizard = mock_wizard([])

paths_cli/tests/wizard/test_two_state_tps.py

@@ -0,0 +1,25 @@
+import pytest
+import mock
+
+from paths_cli.tests.wizard.mock_wizard import mock_wizard
+
+from paths_cli.wizard.two_state_tps import two_state_tps
+
+def mock_tps_scheme(wizard, network):
+    wizard.say(str(network.__class__))
+    return "this would be a scheme"
+
+@mock.patch('paths_cli.wizard.two_state_tps.tps_scheme',
+            new=mock_tps_scheme)
+def test_two_state_tps(tps_network):
+    wizard = mock_wizard([])
+    state_A = tps_network.initial_states[0]
+    state_B = tps_network.final_states[0]
+    with mock.patch('paths_cli.wizard.two_state_tps.volumes',
+                    new=mock.Mock(side_effect=[state_A, state_B])):
+        scheme = two_state_tps(wizard)
+    assert scheme == "this would be a scheme"
+    assert "network.TPSNetwork" in wizard.console.log_text
+
+
+

paths_cli/wizard/core.py

@@ -1,9 +1,9 @@
 import random
-from paths_cli.wizard.joke import name_joke
 from paths_cli.wizard.tools import a_an
 
 from collections import namedtuple
 
+WIZARD_STORE_NAMES = ['engines', 'cvs', 'states', 'networks', 'schemes']
 WizardSay = namedtuple("WizardSay", ['msg', 'mode'])
 
 def interpret_req(req):

paths_cli/wizard/two_state_tps.py

@@ -1,8 +1,8 @@
+from paths_cli.wizard.volumes import volumes
+from paths_cli.wizard.tps import tps_scheme
 from paths_cli.wizard.steps import (
     SINGLE_ENGINE_STEP, CVS_STEP, WizardStep
 )
-from paths_cli.wizard.volumes import volumes
-from paths_cli.wizard.tps import tps_scheme
 from paths_cli.wizard.wizard import Wizard
 
 def two_state_tps(wizard, fixed_length=False):