Add tests

Author: code-yeongyu

aishell/utils/make_executable_command.py

@@ -8,5 +8,5 @@ def make_executable_command(command: str) -> str:
     if command.endswith('`'):
         command = command[:-1]
     command = command.strip()
-    command = command.split('User: ')[0]
+    command = command.split('User: ')[-1]
     return command

aishell/utils/test/__init__.py

@@ -0,0 +1,25 @@
+from aishell.utils import make_executable_command
+
+
+def test_make_executable_command_starts_and_ends_with_newline():
+    command = '\nls -l\n'
+    expected_output = 'ls -l'
+    assert make_executable_command(command) == expected_output
+
+
+def test_make_executable_command_starts_and_ends_with_backtick():
+    command = '`python my_script.py`\n'
+    expected_output = 'python my_script.py'
+    assert make_executable_command(command) == expected_output
+
+
+def test_make_executable_command_starts_with_user():
+    command = 'User: cd ..'
+    expected_output = 'cd ..'
+    assert make_executable_command(command) == expected_output
+
+
+def test_make_executable_command_no_leading_or_trailing_whitespace():
+    command = 'git clone https://github.com/my_repo.git'
+    expected_output = 'git clone https://github.com/my_repo.git'
+    assert make_executable_command(command) == expected_output