move _handle_kind to component

xwakes/basewake.py

@@ -84,18 +84,6 @@ def __init__(self, components):
         self.components = components
 
 
-def _handle_kind(kind):
-    if isinstance(kind, str):
-        kind = [kind]
-
-    if isinstance(kind, (list, tuple)):
-        kind = {kk: 1.0 for kk in kind}
-
-    assert hasattr(kind, 'keys')
-
-    return kind
-
-
 def _expand_components(components):
     _expanded_components = []
     for cc in components:

xwakes/resonator.py

@@ -1,6 +1,7 @@
 from typing import Tuple
 
-from .basewake import BaseWake, _handle_kind
+from .basewake import BaseWake
+from .wit.component import _handle_kind
 from .wit import ComponentResonator
 
 

xwakes/thick_resistive_wall.py

@@ -1,6 +1,7 @@
 from typing import Tuple
 
-from .basewake import BaseWake, _handle_kind
+from .basewake import BaseWake
+from .wit.component import _handle_kind
 from .wit import ComponentClassicThickWall
 
 

xwakes/wit/component.py

@@ -41,6 +41,18 @@
 }
 
 
+def _handle_kind(kind):
+    if isinstance(kind, str):
+        kind = [kind]
+
+    if isinstance(kind, (list, tuple)):
+        kind = {kk: 1.0 for kk in kind}
+
+    assert hasattr(kind, 'keys')
+
+    return kind
+
+
 def mix_fine_and_rough_sampling(start: float, stop: float, rough_points: int,
                                 fine_points: int, rois: List[Tuple[float, float]]):
     """