tests for inline comments in py

Author: ManfredHair

tests/analyze/test_inline_comments.py

@@ -0,0 +1,87 @@
+import os
+from spice.analyzers.count_inline_comments import count_inline_comments
+
+# Get path to the sample files
+SAMPLES_DIR = os.path.join(os.path.dirname(__file__), "..", "sample-code")
+PY_SAMPLE = os.path.join(SAMPLES_DIR, "example.py")
+JS_SAMPLE = os.path.join(SAMPLES_DIR, "example.js")
+GO_SAMPLE = os.path.join(SAMPLES_DIR, "example.go")
+RB_SAMPLE = os.path.join(SAMPLES_DIR, "example.rb")
+
+def test_count_inline_comments_python():
+    """Test counting inline comments in Python code."""
+    count = count_inline_comments(PY_SAMPLE)
+    assert count == 2, f"Expected 2 inline comments in Python sample, got {count}"
+
+def test_count_inline_comments_javascript():
+    """Test counting inline comments in JavaScript code."""
+    count = count_inline_comments(JS_SAMPLE)
+    assert count == 2, f"Expected 2 inline comments in JavaScript sample, got {count}"
+
+def test_count_inline_comments_go():
+    """Test counting inline comments in Go code."""
+    count = count_inline_comments(GO_SAMPLE)
+    assert count == 1, f"Expected 1 inline comment in Go sample, got {count}"
+
+def test_count_inline_comments_ruby():
+    """Test counting inline comments in Ruby code."""
+    count = count_inline_comments(RB_SAMPLE)
+    assert count == 1, f"Expected 1 inline comment in Ruby sample, got {count}"
+
+def test_count_inline_comments_nonexistent_file():
+    """Test counting inline comments in a nonexistent file."""
+    try:
+        count_inline_comments("nonexistent_file.py")
+        assert False, "Expected FileNotFoundError for nonexistent file"
+    except FileNotFoundError:
+        # Expected behavior
+        pass
+
+def test_count_inline_comments_empty_string():
+    """Test counting inline comments in an empty file."""
+    # Create a temporary empty file
+    import tempfile
+    with tempfile.NamedTemporaryFile(suffix=".py", delete=False) as temp:
+        temp_name = temp.name
+    
+    try:
+        count = count_inline_comments(temp_name)
+        assert count == 0, f"Expected 0 inline comments in empty file, got {count}"
+    finally:
+        # Clean up the temporary file
+        os.unlink(temp_name)
+        
+def test_count_inline_comments_with_only_inline_comments():
+    """Test counting inline comments in a file with only inline comments."""
+    # Create a temporary file with only inline comments
+    import tempfile
+    with tempfile.NamedTemporaryFile(suffix=".py", delete=False) as temp:
+        temp.write(b"x = 5  # This is an inline comment\n")
+        temp.write(b"y = 10  # This is another inline comment\n")
+        temp.write(b"z = 15  # This is yet another inline comment\n")
+        temp_name = temp.name
+    
+    try:
+        count = count_inline_comments(temp_name)
+        assert count == 3, f"Expected 3 inline comments in file with only inline comments, got {count}"
+    finally:
+        # Clean up the temporary file
+        os.unlink(temp_name)
+        
+def test_count_inline_comments_with_mixed_comments():
+    """Test counting inline comments in a file with both regular and inline comments."""
+    # Create a temporary file with mixed comments
+    import tempfile
+    with tempfile.NamedTemporaryFile(suffix=".py", delete=False) as temp:
+        temp.write(b"# This is a regular comment\n")
+        temp.write(b"x = 5  # This is an inline comment\n")
+        temp.write(b"# This is another regular comment\n")
+        temp.write(b"y = 10  # This is another inline comment\n")
+        temp_name = temp.name
+    
+    try:
+        count = count_inline_comments(temp_name)
+        assert count == 2, f"Expected 2 inline comments in file with mixed comments, got {count}"
+    finally:
+        # Clean up the temporary file
+        os.unlink(temp_name)