make common test hooks module

Author: rmorshea

tests/test_core/test_layout.py

@@ -20,6 +20,7 @@
     capture_idom_logs,
 )
 from idom.utils import Ref
+from tests.tooling.hooks import use_toggle
 
 
 @pytest.fixture(autouse=True)
@@ -494,11 +495,6 @@ def SomeComponent():
     )
 
 
-def use_toggle(init=False):
-    state, set_state = idom.hooks.use_state(init)
-    return state, lambda: set_state(lambda old: not old)
-
-
 async def test_model_key_preserves_callback_identity_for_common_elements(caplog):
     called_good_trigger = idom.Ref(False)
     good_handler = StaticEventHandler()
@@ -813,18 +809,13 @@ def Child(state):
             await layout.render()
 
 
-def use_toggle():
-    state, set_state = idom.hooks.use_state(True)
-    return state, lambda: set_state(not state)
-
-
 async def test_elements_and_components_with_the_same_key_can_be_interchanged():
     set_toggle = idom.Ref()
     effects = []
 
     @idom.component
     def Root():
-        toggle, set_toggle.current = use_toggle()
+        toggle, set_toggle.current = use_toggle(True)
         if toggle:
             return SomeComponent("x")
         else:

tests/test_html.py

@@ -1,26 +1,17 @@
 import pytest
 
-from idom import component, config, html, use_state
+from idom import component, config, html
 from idom.testing import DisplayFixture, poll
 from idom.utils import Ref
-
-
-def use_toggle(initial=True):
-    state, set_state = use_state(initial)
-    return state, lambda: set_state(not state)
-
-
-def use_counter(initial_value):
-    state, set_state = use_state(initial_value)
-    return state, lambda: set_state(state + 1)
+from tests.tooling.hooks import use_counter, use_toggle
 
 
 async def test_script_mount_unmount(display: DisplayFixture):
     toggle_is_mounted = Ref()
 
     @component
     def Root():
-        is_mounted, toggle_is_mounted.current = use_toggle()
+        is_mounted, toggle_is_mounted.current = use_toggle(True)
         return html.div(
             html.div({"id": "mount-state", "data-value": False}),
             HasScript() if is_mounted else html.div(),

tests/tooling/hooks.py

@@ -0,0 +1,11 @@
+from idom import use_state
+
+
+def use_toggle(init=False):
+    state, set_state = use_state(init)
+    return state, lambda: set_state(lambda old: not old)
+
+
+def use_counter(initial_value):
+    state, set_state = use_state(initial_value)
+    return state, lambda: set_state(state + 1)