1 files changed, 12 insertions, 12 deletions

diff --git a/lib/python/qmk/tests/test_qmk_keymap.py b/lib/python/qmk/tests/test_qmk_keymap.py
index 7ef708e0d..f1ecf2937 100644
--- a/lib/python/qmk/tests/test_qmk_keymap.py
+++ b/lib/python/qmk/tests/test_qmk_keymap.py
@@ -1,33 +1,33 @@
 import qmk.keymap
 
 
-def test_template_onekey_proton_c():
-    templ = qmk.keymap.template('handwired/onekey/proton_c')
+def test_template_c_onekey_proton_c():
+    templ = qmk.keymap.template_c('handwired/onekey/proton_c')
     assert templ == qmk.keymap.DEFAULT_KEYMAP_C
 
 
-def test_template_onekey_proton_c_json():
-    templ = qmk.keymap.template('handwired/onekey/proton_c', type='json')
+def test_template_json_onekey_proton_c():
+    templ = qmk.keymap.template_json('handwired/onekey/proton_c')
     assert templ == {'keyboard': 'handwired/onekey/proton_c'}
 
 
-def test_template_onekey_pytest():
-    templ = qmk.keymap.template('handwired/onekey/pytest')
+def test_template_c_onekey_pytest():
+    templ = qmk.keymap.template_c('handwired/onekey/pytest')
     assert templ == '#include QMK_KEYBOARD_H\nconst uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {__KEYMAP_GOES_HERE__};\n'
 
 
-def test_template_onekey_pytest_json():
-    templ = qmk.keymap.template('handwired/onekey/pytest', type='json')
+def test_template_json_onekey_pytest():
+    templ = qmk.keymap.template_json('handwired/onekey/pytest')
     assert templ == {'keyboard': 'handwired/onekey/pytest', "documentation": "This file is a keymap.json file for handwired/onekey/pytest"}
 
 
-def test_generate_onekey_pytest():
-    templ = qmk.keymap.generate('handwired/onekey/pytest', 'LAYOUT', [['KC_A']])
+def test_generate_c_onekey_pytest():
+    templ = qmk.keymap.generate_c('handwired/onekey/pytest', 'LAYOUT', [['KC_A']])
     assert templ == '#include QMK_KEYBOARD_H\nconst uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {\t[0] = LAYOUT(KC_A)};\n'
 
 
-def test_generate_onekey_pytest_json():
-    templ = qmk.keymap.generate('handwired/onekey/pytest', 'LAYOUT', [['KC_A']], type='json', keymap='default')
+def test_generate_json_onekey_pytest():
+    templ = qmk.keymap.generate_json('default', 'handwired/onekey/pytest', 'LAYOUT', [['KC_A']])
     assert templ == {"keyboard": "handwired/onekey/pytest", "documentation": "This file is a keymap.json file for handwired/onekey/pytest", "keymap": "default", "layout": "LAYOUT", "layers": [["KC_A"]]}