Fix test suite and improve code stability (#6)
* Improved test coverage for audio_feedback, main, and tray_indicator modules * Fix test suite and improve code stability * Black formatter changes * Fixing import sorting
This commit is contained in:
Jatin K Malik
+213
-17
@@ -1,5 +1,5 @@
|
||||
"""
|
||||
Tests for the CommandProcessor component.
|
||||
Tests for the command processor module.
|
||||
"""
|
||||
|
||||
import unittest
|
||||
@@ -11,25 +11,221 @@ from src.speech_recognition.command_processor import CommandProcessor
|
||||
|
||||
|
||||
class TestCommandProcessor(unittest.TestCase):
|
||||
"""Test cases for the command processor."""
|
||||
"""Test cases for command processor functionality."""
|
||||
|
||||
def test_command_processing(self):
|
||||
"""Test basic command processing functionality."""
|
||||
# Create processor
|
||||
processor = CommandProcessor()
|
||||
def setUp(self):
|
||||
"""Set up for tests."""
|
||||
self.processor = CommandProcessor()
|
||||
|
||||
# Test text commands (punctuation, etc.)
|
||||
text, actions = processor.process_text("add a period to the end")
|
||||
assert text == "add a . to the end"
|
||||
assert actions == []
|
||||
def test_initialization(self):
|
||||
"""Test initialization of command processor."""
|
||||
# Verify command dictionaries are initialized
|
||||
self.assertTrue(hasattr(self.processor, "text_commands"))
|
||||
self.assertTrue(hasattr(self.processor, "action_commands"))
|
||||
self.assertTrue(hasattr(self.processor, "format_commands"))
|
||||
|
||||
# Verify dictionaries have expected entries
|
||||
self.assertIn("new line", self.processor.text_commands)
|
||||
self.assertIn("period", self.processor.text_commands)
|
||||
self.assertIn("delete that", self.processor.action_commands)
|
||||
self.assertIn("capitalize", self.processor.format_commands)
|
||||
|
||||
# Verify regex patterns are compiled
|
||||
self.assertTrue(hasattr(self.processor, "text_cmd_regex"))
|
||||
self.assertTrue(hasattr(self.processor, "action_cmd_regex"))
|
||||
self.assertTrue(hasattr(self.processor, "format_cmd_regex"))
|
||||
|
||||
def test_text_command_processing(self):
|
||||
"""Test processing of text commands."""
|
||||
# Test common text commands
|
||||
test_cases = [
|
||||
("new line", "\n", []),
|
||||
("this is a new paragraph", "this is a \n\n", []),
|
||||
("end of sentence period", "end of sentence.", []),
|
||||
("add a comma here", "add a, here", []),
|
||||
("use question mark", "use?", []),
|
||||
("exclamation mark test", "! test", []),
|
||||
("semicolon example", "; example", []),
|
||||
("testing colon usage", "testing:", []),
|
||||
("dash separator", "- separator", []),
|
||||
("hyphen example", "- example", []),
|
||||
("underscore value", "_ value", []),
|
||||
("quote example", '" example', []),
|
||||
("single quote test", "' test", []),
|
||||
("open parenthesis content close parenthesis", "( content)", []),
|
||||
("open bracket item close bracket", "[ item]", []),
|
||||
("open brace code close brace", "{ code}", []),
|
||||
]
|
||||
|
||||
for input_text, expected_output, _ in test_cases:
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, expected_output)
|
||||
self.assertEqual(actions, [])
|
||||
|
||||
def test_action_command_processing(self):
|
||||
"""Test processing of action commands."""
|
||||
# Test action commands
|
||||
text, actions = processor.process_text("delete that please")
|
||||
# Adjust expected text to match actual behavior
|
||||
assert text == "please"
|
||||
assert actions == ["delete_last"]
|
||||
test_cases = [
|
||||
("delete that", "", ["delete_last"]),
|
||||
("scratch that previous text", " previous text", ["delete_last"]),
|
||||
("undo my last change", " my last change", ["undo"]),
|
||||
("redo that edit", " that edit", ["redo"]),
|
||||
("select all text", " text", ["select_all"]),
|
||||
("select line of code", " of code", ["select_line"]),
|
||||
("select word here", " here", ["select_word"]),
|
||||
("select paragraph content", " content", ["select_paragraph"]),
|
||||
("cut this selection", " this selection", ["cut"]),
|
||||
("copy this text", " this text", ["copy"]),
|
||||
("paste here", " here", ["paste"]),
|
||||
# Test multiple actions
|
||||
("select all then copy", " then", ["select_all", "copy"]),
|
||||
]
|
||||
|
||||
for input_text, expected_output, expected_actions in test_cases:
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, expected_output)
|
||||
self.assertEqual(actions, expected_actions)
|
||||
|
||||
def test_format_command_processing(self):
|
||||
"""Test processing of formatting commands."""
|
||||
# Test format commands
|
||||
text, actions = processor.process_text("capitalize hello")
|
||||
assert text == "Hello"
|
||||
assert actions == []
|
||||
test_cases = [
|
||||
("capitalize word", "Word", []),
|
||||
("uppercase letters", "LETTERS", []),
|
||||
("all caps example", "EXAMPLE", []),
|
||||
("lowercase TEXT", "text", []),
|
||||
# Test with text before the command
|
||||
("make this capitalize next", "make this Next", []),
|
||||
]
|
||||
|
||||
for input_text, expected_output, expected_actions in test_cases:
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, expected_output)
|
||||
self.assertEqual(actions, expected_actions)
|
||||
|
||||
def test_combined_commands(self):
|
||||
"""Test combinations of different command types."""
|
||||
test_cases = [
|
||||
# Text + Action
|
||||
("new line then delete that", "", ["delete_last"]),
|
||||
# Format + Text
|
||||
("capitalize name period", "Name.", []),
|
||||
# Action + Format
|
||||
("select all then capitalize text", " then Text", ["select_all"]),
|
||||
# Complex combination
|
||||
(
|
||||
"capitalize name comma new line select paragraph",
|
||||
"Name,\n",
|
||||
["select_paragraph"],
|
||||
),
|
||||
]
|
||||
|
||||
for input_text, expected_output, expected_actions in test_cases:
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, expected_output)
|
||||
self.assertEqual(actions, expected_actions)
|
||||
|
||||
def test_empty_input(self):
|
||||
"""Test handling of empty input."""
|
||||
result, actions = self.processor.process_text("")
|
||||
self.assertEqual(result, "")
|
||||
self.assertEqual(actions, [])
|
||||
|
||||
result, actions = self.processor.process_text(None)
|
||||
self.assertEqual(result, "")
|
||||
self.assertEqual(actions, [])
|
||||
|
||||
def test_no_commands(self):
|
||||
"""Test text with no commands."""
|
||||
input_text = "This is just regular text with no commands."
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, input_text)
|
||||
self.assertEqual(actions, [])
|
||||
|
||||
def test_partial_command_matches(self):
|
||||
"""Test text with partial command matches."""
|
||||
# Words that are substrings of commands should not be processed
|
||||
test_cases = [
|
||||
("periodic review", "periodic review", []), # shouldn't match "period"
|
||||
("newcomer", "newcomer", []), # shouldn't match "new"
|
||||
("paramount", "paramount", []), # shouldn't match "paragraph"
|
||||
]
|
||||
|
||||
for input_text, expected_output, expected_actions in test_cases:
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, expected_output)
|
||||
self.assertEqual(actions, expected_actions)
|
||||
|
||||
def test_case_insensitivity(self):
|
||||
"""Test that command matching is case-insensitive."""
|
||||
test_cases = [
|
||||
("NEW LINE", "\n", []),
|
||||
("Period", ".", []),
|
||||
("DELETE THAT", "", ["delete_last"]),
|
||||
("Capitalize word", "Word", []),
|
||||
]
|
||||
|
||||
for input_text, expected_output, expected_actions in test_cases:
|
||||
result, actions = self.processor.process_text(input_text)
|
||||
self.assertEqual(result, expected_output)
|
||||
self.assertEqual(actions, expected_actions)
|
||||
|
||||
def test_multiple_format_modifiers(self):
|
||||
"""Test multiple format modifiers on a single word."""
|
||||
# Reset command processor to ensure active_formats is empty
|
||||
self.processor = CommandProcessor()
|
||||
|
||||
# Apply multiple formatting commands
|
||||
result, _ = self.processor.process_text("capitalize all caps text")
|
||||
|
||||
# The second format command overrides the first, resulting in all uppercase
|
||||
self.assertEqual(result, "TEXT")
|
||||
|
||||
def test_format_with_no_target_word(self):
|
||||
"""Test format command with no following word."""
|
||||
result, _ = self.processor.process_text("capitalize")
|
||||
self.assertEqual(result, "")
|
||||
|
||||
# The active_formats should be cleared even if no word was formatted
|
||||
self.assertEqual(self.processor.active_formats, set())
|
||||
|
||||
def test_whitespace_handling(self):
|
||||
"""Test handling of whitespace in command processing."""
|
||||
# Test with extra spaces
|
||||
result, _ = self.processor.process_text("new line test")
|
||||
self.assertEqual(result, "\n test")
|
||||
|
||||
# Test with leading/trailing spaces
|
||||
result, _ = self.processor.process_text(" period ")
|
||||
self.assertEqual(result, ".")
|
||||
|
||||
# Test with mixed whitespace
|
||||
result, _ = self.processor.process_text(" capitalize word new line ")
|
||||
self.assertEqual(result, "Word \n")
|
||||
|
||||
def test_regex_compilation(self):
|
||||
"""Test the regex pattern compilation."""
|
||||
self.processor._compile_patterns()
|
||||
|
||||
# Test regex patterns match correctly
|
||||
self.assertTrue(self.processor.text_cmd_regex.search("new line"))
|
||||
self.assertTrue(self.processor.text_cmd_regex.search("this is a period"))
|
||||
self.assertTrue(self.processor.action_cmd_regex.search("delete that"))
|
||||
self.assertTrue(self.processor.format_cmd_regex.search("capitalize this"))
|
||||
|
||||
# Test non-matches
|
||||
self.assertFalse(self.processor.text_cmd_regex.search("newline")) # no space
|
||||
self.assertFalse(
|
||||
self.processor.action_cmd_regex.search("deletion")
|
||||
) # not a command
|
||||
|
||||
def test_compile_patterns_method(self):
|
||||
"""Test the _compile_patterns method directly."""
|
||||
# Add a new command to test recompilation
|
||||
self.processor.text_commands["test command"] = "TEST"
|
||||
|
||||
# Recompile patterns
|
||||
self.processor._compile_patterns()
|
||||
|
||||
# Verify new command is in the pattern
|
||||
self.assertTrue(self.processor.text_cmd_regex.search("test command"))
|
||||
|
||||
Reference in New Issue
Block a user