Change to be an extension

tests/keyboard_test.py

@@ -16,7 +16,12 @@ class DigitalInOut(Mock):
 
 class KeyboardTest:
     def __init__(
-        self, modules, keymap, keyboard_debug_enabled=False, debug_enabled=False
+        self,
+        modules,
+        keymap,
+        keyboard_debug_enabled=False,
+        debug_enabled=False,
+        extensions={},
     ):
         self.debug_enabled = debug_enabled
 
@@ -24,6 +29,7 @@ class KeyboardTest:
         self.keyboard.debug_enabled = keyboard_debug_enabled
 
         self.keyboard.modules = modules
+        self.keyboard.extensions = extensions
 
         self.pins = tuple(DigitalInOut() for k in keymap[0])
 

tests/test_kmk_extension_keymap_string_keynames.py

@@ -0,0 +1,25 @@
+import unittest
+
+from kmk.extensions.keymap_string_keynames import keymap_string_keynames
+from kmk.keys import KC
+from tests.keyboard_test import KeyboardTest
+
+
+class Test_extension_keymap_string_keynames(unittest.TestCase):
+    def test_basic_kmk_keyboard_replace_string_primary_name(self):
+        keyboard = KeyboardTest(
+            [], [['1', '2', '3', '4']], extensions={keymap_string_keynames()}
+        )
+
+        keyboard.test('Simple key press', [(0, True), (0, False)], [{KC.N1}, {}])
+
+    def test_basic_kmk_keyboard_replace_string_secondary_name(self):
+        keyboard = KeyboardTest(
+            [], [['N1', 'N2', 'N3', 'N4']], extensions={keymap_string_keynames()}
+        )
+
+        keyboard.test('Simple key press', [(0, True), (0, False)], [{KC.N1}, {}])
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_kmk_keyboard.py

@@ -10,20 +10,6 @@ class TestKmkKeyboard(unittest.TestCase):
 
         keyboard.test('Simple key press', [(0, True), (0, False)], [{KC.N1}, {}])
 
-    def test_basic_kmk_keyboard_replace_string_primary_name(self):
-        keyboard = KeyboardTest([], [['1', '2', '3', '4']])
-
-        keyboard.test('Simple key press', [(0, True), (0, False)], [{KC.N1}, {}])
-
-    def test_basic_kmk_keyboard_replace_string_secondary_name(self):
-        keyboard = KeyboardTest([], [['N1', 'N2', 'N3', 'N4']])
-
-        keyboard.test('Simple key press', [(0, True), (0, False)], [{KC.N1}, {}])
-
-    def test_basic_kmk_keyboard_unknown_replacement_string(self):
-        with self.assertRaises(ValueError):
-            KeyboardTest([], [['UNKNOWN1', 'UNKNOWN2', 'UNKNOWN3', 'UNKNOWN4']])
-
 
 if __name__ == '__main__':
     unittest.main()