test: add more test cases

Author: wislertt

.amazonq/rules/test-case-enhancement.md

@@ -2,6 +2,20 @@
 
 ## Assistant Workflow for Adding Comprehensive Test Cases
 
+### Automated Workflow (Recommended)
+
+For single problem enhancement:
+
+1. **Use automation script**: `./.amazonq/rules/test-case-enhancement/scripts/enhance_single_problem.sh <problem_name>`
+2. **Verify results**: `./.amazonq/rules/test-case-enhancement/scripts/verify_enhancement.sh <problem_name>`
+
+For batch enhancement:
+
+1. **Find problems**: `./.amazonq/rules/test-case-enhancement/scripts/find_problems_needing_enhancement.sh`
+2. **Batch enhance**: `./.amazonq/rules/test-case-enhancement/scripts/enhance_batch_problems.sh`
+
+### Manual Workflow (Fallback)
+
 When user requests to enhance test cases for a problem, the assistant will:
 
 ### 1. Problem Resolution (Priority Order)
@@ -98,7 +112,45 @@ When user requests to enhance test cases for a problem, the assistant will:
 - **NEVER include custom solution classes** in test_imports - only import the main solution class specified in solution_class_name
 - **PRESERVE existing solution class parametrization** - if original test had multiple solution classes, restore them after JSON regeneration
 
-## Commands Reference
+## Automation Scripts 🤖
+
+**RECOMMENDED APPROACH**: Use prebuilt scripts in `.amazonq/rules/test-case-enhancement/scripts/` to streamline the enhancement process:
+
+### Find Problems Needing Enhancement
+
+```bash
+# Find problems with ≤10 test cases (default)
+./.amazonq/rules/test-case-enhancement/scripts/find_problems_needing_enhancement.sh
+
+# Custom threshold and max results
+./.amazonq/rules/test-case-enhancement/scripts/find_problems_needing_enhancement.sh 8 15
+```
+
+### Enhance Single Problem
+
+```bash
+# Enhance specific problem (handles full workflow automatically)
+./.amazonq/rules/test-case-enhancement/scripts/enhance_single_problem.sh <problem_name>
+```
+
+### Batch Enhancement
+
+```bash
+# Enhance up to 10 problems with ≤10 test cases (default)
+./.amazonq/rules/test-case-enhancement/scripts/enhance_batch_problems.sh
+
+# Custom batch size and threshold
+./.amazonq/rules/test-case-enhancement/scripts/enhance_batch_problems.sh 5 8
+```
+
+### Verify Enhancement Results
+
+```bash
+# Verify enhancement was successful
+./.amazonq/rules/test-case-enhancement/scripts/verify_enhancement.sh <problem_name>
+```
+
+## Manual Commands Reference
 
 ```bash
 # Find problems needing more test cases
@@ -129,3 +181,33 @@ make p-lint PROBLEM={problem_name}
 - Original solution code preserved and working
 - JSON template updated for future regeneration
 - Clean final state with no temporary files
+
+## Automation Benefits
+
+Using the prebuilt scripts provides several advantages:
+
+### ⚡ **Speed & Efficiency**
+
+- **Single command execution** - No need to remember complex multi-step workflows
+- **Batch processing** - Enhance multiple problems simultaneously
+- **Automatic cleanup** - No manual file management required
+
+### 🛡️ **Safety & Reliability**
+
+- **Automatic backups** - Original code always preserved
+- **Error handling** - Graceful failure recovery
+- **Validation checks** - Syntax and structure verification
+
+### 📊 **Visibility & Tracking**
+
+- **Progress reporting** - Real-time status updates
+- **Result verification** - Automatic test case counting
+- **Success metrics** - Clear pass/fail indicators
+
+### 🔄 **Consistency**
+
+- **Standardized process** - Same workflow every time
+- **Quality assurance** - Built-in lint and validation checks
+- **Reproducible results** - Eliminates human error
+
+**Recommendation**: Always use automation scripts for test case enhancement unless debugging specific issues requires manual intervention.

.amazonq/rules/test-case-enhancement/scripts/README.md

@@ -0,0 +1,76 @@
+# Test Case Enhancement Automation Scripts
+
+This directory contains prebuilt scripts to streamline the test case enhancement process for LeetCode problems.
+
+## Scripts Overview
+
+### 🔍 `find_problems_needing_enhancement.sh`
+
+**Purpose**: Find problems with insufficient test cases
+**Usage**: `bash find_problems_needing_enhancement.sh [threshold] [max_results]`
+**Example**: `bash find_problems_needing_enhancement.sh 10 5`
+
+### 🚀 `enhance_single_problem.sh`
+
+**Purpose**: Enhance test cases for a single problem (full automated workflow)
+**Usage**: `bash enhance_single_problem.sh <problem_name>`
+**Example**: `bash enhance_single_problem.sh two_sum`
+
+### 📦 `enhance_batch_problems.sh`
+
+**Purpose**: Enhance test cases for multiple problems in batch
+**Usage**: `bash enhance_batch_problems.sh [max_problems] [threshold]`
+**Example**: `bash enhance_batch_problems.sh 5 10`
+
+### ✅ `verify_enhancement.sh`
+
+**Purpose**: Verify that test case enhancement was successful
+**Usage**: `bash verify_enhancement.sh <problem_name>`
+**Example**: `bash verify_enhancement.sh two_sum`
+
+## Typical Workflow
+
+1. **Find problems needing enhancement**:
+
+    ```bash
+    bash .amazonq/rules/test-case-enhancement/scripts/find_problems_needing_enhancement.sh
+    ```
+
+2. **Enhance problems** (choose one):
+
+    ```bash
+    # Single problem
+    bash .amazonq/rules/test-case-enhancement/scripts/enhance_single_problem.sh <problem_name>
+
+    # Batch enhancement
+    bash .amazonq/rules/test-case-enhancement/scripts/enhance_batch_problems.sh
+    ```
+
+3. **Verify results**:
+    ```bash
+    bash .amazonq/rules/test-case-enhancement/scripts/verify_enhancement.sh <problem_name>
+    ```
+
+## What the Scripts Do
+
+### Automated Process
+
+1. **Backup** original problem structure
+2. **Regenerate** from enhanced JSON templates
+3. **Lint check** to ensure code quality
+4. **Preserve** enhanced test files
+5. **Restore** original solution code
+6. **Verify** final state
+
+### Safety Features
+
+- Automatic backups before any changes
+- Rollback capability if errors occur
+- Syntax validation of generated tests
+- Comprehensive error handling
+
+## Requirements
+
+- All scripts must be run from the repository root directory
+- Requires `poetry`, `make`, and standard Unix tools
+- JSON templates must be updated with enhanced test cases before running scripts

.amazonq/rules/test-case-enhancement/scripts/enhance_batch_problems.sh

@@ -0,0 +1,75 @@
+#!/bin/bash
+
+# Batch Test Case Enhancement Script
+# Usage: ./enhance_batch_problems.sh [max_problems]
+
+set -e
+
+MAX_PROBLEMS="${1:-10}"
+THRESHOLD="${2:-10}"
+
+echo "🚀 Starting batch test case enhancement"
+echo "📊 Finding problems with ≤$THRESHOLD test cases (max: $MAX_PROBLEMS)"
+
+# Get list of problems needing enhancement
+PROBLEMS=$(poetry run python .templates/check_test_cases.py --threshold="$THRESHOLD" --max="$MAX_PROBLEMS" | grep "\.json:" | cut -d':' -f1 | sed 's/\.json$//')
+
+if [ -z "$PROBLEMS" ]; then
+    echo "✅ No problems found needing test case enhancement"
+    exit 0
+fi
+
+echo "📋 Problems to enhance:"
+echo "$PROBLEMS" | sed 's/^/  - /'
+
+# Create backup directory
+mkdir -p .cache/leetcode
+
+# Process each problem
+SUCCESS_COUNT=0
+TOTAL_COUNT=0
+
+for PROBLEM in $PROBLEMS; do
+    TOTAL_COUNT=$((TOTAL_COUNT + 1))
+    echo ""
+    echo "🔄 Processing problem $TOTAL_COUNT: $PROBLEM"
+
+    # Backup original
+    if [ -d "leetcode/$PROBLEM" ]; then
+        cp -r "leetcode/$PROBLEM" ".cache/leetcode/"
+        echo "  📦 Backed up original"
+    else
+        echo "  ❌ Problem directory not found: leetcode/$PROBLEM"
+        continue
+    fi
+
+    # Regenerate from JSON
+    if make p-gen PROBLEM="$PROBLEM" FORCE=1 > /dev/null 2>&1; then
+        echo "  🔄 Regenerated from JSON"
+    else
+        echo "  ❌ Failed to regenerate from JSON"
+        continue
+    fi
+
+    # Preserve enhanced test and restore structure
+    cp "leetcode/$PROBLEM/test_solution.py" ".temp_test_$PROBLEM.py"
+    rm -rf "leetcode/$PROBLEM"
+    cp -r ".cache/leetcode/$PROBLEM" "leetcode/"
+    cp ".temp_test_$PROBLEM.py" "leetcode/$PROBLEM/test_solution.py"
+    rm ".temp_test_$PROBLEM.py"
+
+    SUCCESS_COUNT=$((SUCCESS_COUNT + 1))
+    echo "  ✅ Enhanced successfully"
+done
+
+# Cleanup
+rm -rf .cache/leetcode
+
+echo ""
+echo "🎉 Batch enhancement completed!"
+echo "📊 Results: $SUCCESS_COUNT/$TOTAL_COUNT problems enhanced successfully"
+
+# Verify results
+echo ""
+echo "📋 Final verification:"
+poetry run python .templates/check_test_cases.py --threshold="$THRESHOLD" --max=5

.amazonq/rules/test-case-enhancement/scripts/enhance_single_problem.sh

@@ -0,0 +1,64 @@
+#!/bin/bash
+
+# Enhanced Test Case Generation Script for Single Problem
+# Usage: ./enhance_single_problem.sh <problem_name>
+
+set -e
+
+PROBLEM_NAME="$1"
+
+if [ -z "$PROBLEM_NAME" ]; then
+    echo "Usage: $0 <problem_name>"
+    exit 1
+fi
+
+echo "🚀 Enhancing test cases for: $PROBLEM_NAME"
+
+# Step 1: Backup original problem
+echo "📦 Backing up original problem..."
+mkdir -p .cache/leetcode
+if [ -d "leetcode/$PROBLEM_NAME" ]; then
+    cp -r "leetcode/$PROBLEM_NAME" ".cache/leetcode/"
+    echo "✅ Backup created"
+else
+    echo "❌ Problem directory not found: leetcode/$PROBLEM_NAME"
+    exit 1
+fi
+
+# Step 2: Regenerate from JSON template
+echo "🔄 Regenerating from enhanced JSON template..."
+make p-gen PROBLEM="$PROBLEM_NAME" FORCE=1
+
+# Step 3: Lint check
+echo "🔍 Running lint check..."
+if make p-lint PROBLEM="$PROBLEM_NAME"; then
+    echo "✅ Lint check passed"
+else
+    echo "❌ Lint check failed - please fix JSON template"
+    exit 1
+fi
+
+# Step 4: Preserve enhanced test file and restore original structure
+echo "🔄 Preserving enhanced tests and restoring original structure..."
+cp "leetcode/$PROBLEM_NAME/test_solution.py" ".temp_enhanced_test.py"
+rm -rf "leetcode/$PROBLEM_NAME"
+cp -r ".cache/leetcode/$PROBLEM_NAME" "leetcode/"
+cp ".temp_enhanced_test.py" "leetcode/$PROBLEM_NAME/test_solution.py"
+rm ".temp_enhanced_test.py"
+
+# Step 5: Verify final state
+echo "✅ Enhancement complete for: $PROBLEM_NAME"
+echo "📊 Test case count verification:"
+python3 -c "
+import re
+with open('leetcode/$PROBLEM_NAME/test_solution.py', 'r') as f:
+    content = f.read()
+    matches = re.findall(r'@pytest\.mark\.parametrize.*?\[(.*?)\]', content, re.DOTALL)
+    if matches:
+        test_cases = matches[0].count('(')
+        print(f'Total test cases: {test_cases}')
+    else:
+        print('Could not count test cases')
+"
+
+echo "🎉 Test case enhancement completed successfully!"

.amazonq/rules/test-case-enhancement/scripts/find_problems_needing_enhancement.sh

@@ -0,0 +1,39 @@
+#!/bin/bash
+
+# Find Problems Needing Test Case Enhancement
+# Usage: ./find_problems_needing_enhancement.sh [threshold] [max_results]
+
+THRESHOLD="${1:-10}"
+MAX_RESULTS="${2:-20}"
+
+echo "🔍 Finding problems with ≤$THRESHOLD test cases (showing max $MAX_RESULTS results)"
+echo ""
+
+# Run the check and format output
+OUTPUT=$(poetry run python .templates/check_test_cases.py --threshold="$THRESHOLD" --max="$MAX_RESULTS")
+
+if echo "$OUTPUT" | grep -q "Files with ≤"; then
+    echo "$OUTPUT"
+    echo ""
+
+    # Extract just the problem names for easy copying
+    PROBLEM_NAMES=$(echo "$OUTPUT" | grep "\.json:" | cut -d':' -f1 | sed 's/\.json$//')
+    PROBLEM_COUNT=$(echo "$PROBLEM_NAMES" | wc -l | tr -d ' ')
+
+    echo "📋 Problem names for batch processing:"
+    echo "$PROBLEM_NAMES" | sed 's/^/  /'
+    echo ""
+    echo "📊 Total problems found: $PROBLEM_COUNT"
+
+    if [ "$PROBLEM_COUNT" -gt 0 ]; then
+        echo ""
+        echo "💡 Quick commands:"
+        echo "  # Enhance all found problems:"
+        echo "  ./.amazonq/rules/test-case-enhancement/scripts/enhance_batch_problems.sh $PROBLEM_COUNT $THRESHOLD"
+        echo ""
+        echo "  # Enhance single problem:"
+        echo "  ./.amazonq/rules/test-case-enhancement/scripts/enhance_single_problem.sh <problem_name>"
+    fi
+else
+    echo "✅ No problems found needing test case enhancement (threshold: $THRESHOLD)"
+fi