feat: move count_test_cases

Author: wislertt

.amazonq/plans/cli-implementation.md

@@ -1,4 +1,4 @@
-# Test Case Enhancement Rules
+# Test Quality Assurance Rules
 
 ## Simple Enhancement Workflow
 
@@ -7,7 +7,7 @@ When user requests test case enhancement or **test reproducibility verification*
 ### 1. Problem Resolution
 
 - Use active file context or user-provided problem name
-- If unclear, run: `poetry run python .templates/check_test_cases.py --threshold=10 --max=1`
+- If unclear, run: `poetry run python -m leetcode_py.tools.check_test_cases --threshold=10 --max=1`
 
 ### 2. Enhancement Process
 
@@ -72,7 +72,13 @@ make p-lint PROBLEM={problem_name}
 
 ```bash
 # Find problems needing enhancement
-poetry run python .templates/check_test_cases.py --threshold=10
+poetry run python -m leetcode_py.tools.check_test_cases --threshold=10
+
+# Check all problems (no limit)
+poetry run python -m leetcode_py.tools.check_test_cases --threshold=10 --max=none
+
+# Check with custom threshold
+poetry run python -m leetcode_py.tools.check_test_cases --threshold=12
 
 # Generate from JSON template (uses lcpy internally)
 make p-gen PROBLEM={problem_name} FORCE=1

.amazonq/plans/test-case-integration.md

@@ -13,7 +13,7 @@ Use an LLM assistant (Cursor, GitHub Copilot Chat, Amazon Q) with the rule files
 
 ### 2. Enhance Test Cases
 
-- Include `.amazonq/rules/test-case-enhancement.md` in your LLM context
+- Include `.amazonq/rules/test-quality-assurance.md` in your LLM context
 - Ask: "Enhance test cases for [problem_name] problem"
 
 ### 3. Improve Helper Classes

.amazonq/rules/test-case-enhancement.md renamed to .amazonq/rules/test-quality-assurance.md

@@ -84,6 +84,11 @@ nb-to-py:
 	@find leetcode -name "*.ipynb" -delete
 	@echo "Conversion complete. All .ipynb files converted to .py and deleted."
 
+# Find problems with few test cases
+check-test-cases:
+	@echo "Checking test case coverage..."
+	poetry run python leetcode_py/tools/check_test_cases.py --threshold=$(or $(THRESHOLD),10) --max=$(or $(MAX),1)
+
 # Generate All Problems - useful for people who fork this repo
 gen-all-problems:
 	@echo "This will DELETE all existing problems and regenerate from JSON templates."

.templates/check_test_cases.py

@@ -163,7 +163,7 @@ If you need more problems beyond Grind 75, use an LLM assistant in your IDE (Cur
 **Required LLM Context**: Include these rule files in your LLM context for automated problem generation and test enhancement:
 
 - [`.amazonq/rules/problem-creation.md`](.amazonq/rules/problem-creation.md) - Complete problem generation workflow
-- [`.amazonq/rules/test-case-enhancement.md`](.amazonq/rules/test-case-enhancement.md) - Test enhancement and reproducibility verification
+- [`.amazonq/rules/test-quality-assurance.md`](.amazonq/rules/test-quality-assurance.md) - Test enhancement and reproducibility verification
 
 **Manual Check**: Find problems needing more test cases:
 

CONTRIBUTING.md

@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+
+import ast
+import json
+from typing import Any
+
+import typer
+
+from leetcode_py.cli.utils.resources import get_problems_json_path
+
+
+def count_test_cases_for_problem(json_data: dict[str, Any]) -> int:
+    """Count total test cases across all test methods for a problem."""
+    total = 0
+
+    # Handle both direct test_methods and nested _test_methods.list
+    test_methods = json_data.get("test_methods", [])
+    if not test_methods and "_test_methods" in json_data:
+        test_methods = json_data["_test_methods"].get("list", [])
+
+    for method in test_methods:
+        test_cases = method.get("test_cases", "")
+        if test_cases.strip():
+            # Parse Python list literal using ast.literal_eval
+            cases_list = ast.literal_eval(test_cases)
+            total += len(cases_list)
+    return total
+
+
+def check_test_cases(
+    threshold: int = typer.Option(
+        10, "--threshold", "-t", help="Show problems with test cases <= threshold"
+    ),
+    max_results: str = typer.Option(
+        "10", "--max", "-m", help="Maximum number of results to show ('none' for no limit)"
+    ),
+) -> None:
+    """Check test case counts in LeetCode problems."""
+    problems_dir = get_problems_json_path()
+    all_problems: list[tuple[str, int]] = []
+
+    for problem_file in problems_dir.glob("*.json"):
+        try:
+            with open(problem_file) as f:
+                data = json.load(f)
+
+            test_count = count_test_cases_for_problem(data)
+            all_problems.append((problem_file.name, test_count))
+        except Exception as e:
+            typer.echo(f"Error reading problem {problem_file.name}: {e}", err=True)
+
+    # Sort by test count
+    all_problems.sort(key=lambda x: x[1])
+
+    # Filter by threshold
+    filtered_problems = [p for p in all_problems if p[1] <= threshold]
+
+    # Apply max results limit
+    if max_results.lower() not in ["none", "null", "-1"]:
+        try:
+            max_count = int(max_results)
+            if max_count > 0:
+                filtered_problems = filtered_problems[:max_count]
+        except ValueError:
+            typer.echo(f"Invalid max_results value: {max_results}", err=True)
+            raise typer.Exit(1)
+
+    typer.echo(f"Problems with ≤{threshold} test cases ({len(filtered_problems)} total):")
+    for problem_name, count in filtered_problems:
+        typer.echo(f"{problem_name}: {count} test cases")
+
+    # Exit with non-zero code if any problems found
+    if filtered_problems:
+        raise typer.Exit(1)
+
+
+app = typer.Typer()
+app.command()(check_test_cases)
+
+if __name__ == "__main__":
+    app()