wcgw 2.8.3__tar.gz → 2.8.5__tar.gz

wcgw-2.8.5/Dockerfile

@@ -0,0 +1,41 @@
+# Generated by https://smithery.ai. See: https://smithery.ai/docs/config#dockerfile
+# Start from a Python base image with the necessary tools
+FROM ghcr.io/astral-sh/uv:python3.12-bookworm-slim AS uv
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the project's pyproject.toml and lock file for dependency installation
+COPY pyproject.toml /app/
+COPY uv.lock /app/
+
+# Enable bytecode compilation and set link mode to copy for dependencies
+ENV UV_COMPILE_BYTECODE=1
+ENV UV_LINK_MODE=copy
+
+# Install dependencies
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv sync --frozen --no-install-project --no-dev --no-editable
+
+# Copy the entire project into the container
+COPY . /app
+
+# Install the project
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv sync --frozen --no-dev --no-editable
+
+# Use a smaller image to run the application
+FROM python:3.12-slim-bookworm
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the installed application from the previous stage
+COPY --from=uv /root/.local /root/.local
+COPY --from=uv --chown=app:app /app/.venv /app/.venv
+
+# Add the virtual environment to the PATH
+ENV PATH="/app/.venv/bin:$PATH"
+
+# Specify the command to run on container start
+ENTRYPOINT ["wcgw_mcp"]

{wcgw-2.8.3 → wcgw-2.8.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 2.8.3
+Version: 2.8.5
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -39,6 +39,7 @@ Description-Content-Type: text/markdown
 [![Mypy strict](https://github.com/rusiaaman/wcgw/actions/workflows/python-types.yml/badge.svg?branch=main)](https://github.com/rusiaaman/wcgw/actions/workflows/python-types.yml)
 [![Build](https://github.com/rusiaaman/wcgw/actions/workflows/python-publish.yml/badge.svg)](https://github.com/rusiaaman/wcgw/actions/workflows/python-publish.yml)
 [![codecov](https://codecov.io/gh/rusiaaman/wcgw/graph/badge.svg)](https://codecov.io/gh/rusiaaman/wcgw)
+[![smithery badge](https://smithery.ai/badge/wcgw)](https://smithery.ai/server/wcgw)
 
 ## Updates
 - [15 Jan 2025] Modes introduced: architect, code-writer, and all powerful wcgw mode. 
@@ -128,7 +129,6 @@ _If there's an error in setting up_
 - Debug the mcp server using `npx @modelcontextprotocol/inspector@0.1.7 uv tool run --from wcgw@latest --python 3.12 wcgw_mcp`
 
 ### Alternative configuration using smithery (npx required)
-[![smithery badge](https://smithery.ai/badge/wcgw)](https://smithery.ai/server/wcgw)
 
 You need to first install uv using homebrew. `brew install uv`
 

{wcgw-2.8.3 → wcgw-2.8.5}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "gapypi@arcfu.com" }]
 name = "wcgw"
-version = "2.8.3"
+version = "2.8.5"
 description = "Shell and coding agent on claude and chatgpt"
 readme = "README.md"
 requires-python = ">=3.11, <3.13"
@@ -42,7 +42,7 @@ packages = ["src/wcgw", "src/mcp_wcgw_fork/src/mcp_wcgw"]
 "src/wcgw" = "wcgw"
 
 [project.scripts]
-wcgw_local = "wcgw:app"
+wcgw_local = "wcgw_cli:app"
 wcgw = "wcgw:listen"
 wcgw_relay = "wcgw.relay.serve:run"
 wcgw_mcp = "wcgw:mcp_server"

{wcgw-2.8.3 → wcgw-2.8.5}/src/wcgw/__init__.py

@@ -1,3 +1,2 @@
-from .client.cli import app
-from .client.tools import run as listen
 from .client.mcp_server import main as mcp_server
+from .client.tools import run as listen

{wcgw-2.8.3 → wcgw-2.8.5}/src/wcgw/client/mcp_server/server.py

@@ -1,15 +1,15 @@
 import importlib
 import json
+import logging
 import os
 from typing import Any
 
-from pydantic import AnyUrl, ValidationError
-
 import mcp_wcgw.server.stdio
 import mcp_wcgw.types as types
 from mcp_wcgw.server import NotificationOptions, Server
 from mcp_wcgw.server.models import InitializationOptions
 from mcp_wcgw.types import Tool as ToolParam
+from pydantic import AnyUrl, ValidationError
 
 from ...types_ import (
     BashCommand,
@@ -35,6 +35,18 @@ COMPUTER_USE_ON_DOCKER_ENABLED = False
 
 server = Server("wcgw")
 
+# Log only time stamp
+logging.basicConfig(level=logging.INFO, format="%(asctime)s: %(message)s")
+logger = logging.getLogger("wcgw")
+
+
+class Console:
+    def print(self, msg: str, *args: Any, **kwargs: Any) -> None:
+        logger.info(msg)
+
+    def log(self, msg: str, *args: Any, **kwargs: Any) -> None:
+        logger.info(msg)
+
 
 @server.list_resources()  # type: ignore
 async def handle_list_resources() -> list[types.Resource]:
@@ -301,7 +313,7 @@ async def main(computer_use: bool) -> None:
     tools.TIMEOUT = SLEEP_TIME_MAX_S
     tools.TIMEOUT_WHILE_OUTPUT = 55
     tools.OUTPUT_WAIT_PATIENCE = 5
-    tools.console = tools.DisableConsole()
+    tools.console = Console()
 
     if computer_use:
         COMPUTER_USE_ON_DOCKER_ENABLED = True

{wcgw-2.8.3 → wcgw-2.8.5}/src/wcgw/client/modes.py

@@ -51,7 +51,7 @@ def code_writer_prompt(
     allowed_commands: Literal["all"] | list[str],
 ) -> str:
     base = """
-You have to run in "code_writer" mode.
+You are now running in "code_writer" mode.
 """
 
     path_prompt = """
@@ -134,7 +134,7 @@ Additional instructions:
 
 
 """
-ARCHITECT_PROMPT = """You have to run in "architect" mode. This means
+ARCHITECT_PROMPT = """You are now running in "architect" mode. This means
 - You are not allowed to edit or update any file. You are not allowed to create any file. 
 - You are not allowed to run any commands that may change disk, system configuration, packages or environment. Only read-only commands are allowed.
 - Only run commands that allows you to explore the repository, understand the system or read anything of relevance. 
@@ -203,7 +203,7 @@ Format the `description` field using Markdown with the following sections.
 - "# Objective" section containing project and task objective.
 - "# All user instructions" section should be provided containing all instructions user shared in the conversation.
 - "# Current status of the task" should be provided containing only what is already achieved, not what's remaining.
-- "# All issues with snippets" section containing snippets of error, traceback, file snippets, commands, etc. But no comments or solutions.
+- "# Pending issues with snippets" section containing snippets of pending errors, traceback, file snippets, commands, etc. But no comments or solutions.
 - Be very verbose in the all issues with snippets section providing as much error context as possible.
 - "# Build and development instructions" section containing instructions to build or run project or run tests, or envrionment related information. Only include what's known. Leave empty if unknown.
 - Any other relevant sections following the above.

{wcgw-2.8.3 → wcgw-2.8.5}/src/wcgw/client/tools.py

@@ -20,6 +20,7 @@ from typing import (
     Literal,
     Optional,
     ParamSpec,
+    Protocol,
     Type,
     TypeVar,
 )
@@ -73,17 +74,13 @@ from .repo_ops.repo_context import get_repo_context
 from .sys_utils import command_run
 
 
-class DisableConsole:
-    def print(self, *args, **kwargs):  # type: ignore
-        pass
+class Console(Protocol):
+    def print(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
 
-    def log(self, *args, **kwargs):  # type: ignore
-        pass
+    def log(self, msg: str, *args: Any, **kwargs: Any) -> None: ...
 
 
-console: rich.console.Console | DisableConsole = rich.console.Console(
-    style="magenta", highlight=False, markup=False
-)
+console: Console = rich.console.Console(style="magenta", highlight=False, markup=False)
 
 TIMEOUT = 5
 TIMEOUT_WHILE_OUTPUT = 20

wcgw-2.8.5/src/wcgw_cli/__init__.py

@@ -0,0 +1 @@
+from .cli import app

{wcgw-2.8.3/src/wcgw/client → wcgw-2.8.5/src/wcgw_cli}/anthropic_client.py

@@ -22,7 +22,17 @@ from anthropic.types import (
 from dotenv import load_dotenv
 from typer import Typer
 
-from ..types_ import (
+from wcgw.client.common import discard_input
+from wcgw.client.memory import load_memory
+from wcgw.client.tools import (
+    DoneFlag,
+    ImageData,
+    default_enc,
+    get_tool_output,
+    initialize,
+    which_tool_name,
+)
+from wcgw.types_ import (
     BashCommand,
     BashInteraction,
     ContextSave,
@@ -36,16 +46,6 @@ from ..types_ import (
     ScreenShot,
     WriteIfEmpty,
 )
-from .common import discard_input
-from .memory import load_memory
-from .tools import (
-    DoneFlag,
-    ImageData,
-    default_enc,
-    get_tool_output,
-    initialize,
-    which_tool_name,
-)
 
 History = list[MessageParam]
 
@@ -290,7 +290,11 @@ Saves provided description and file contents of all the relevant file paths or g
         mode="wcgw",
     )
 
-    with open(os.path.join(os.path.dirname(__file__), "diff-instructions.txt")) as f:
+    with open(
+        os.path.join(
+            os.path.dirname(__file__), "..", "wcgw", "client", "diff-instructions.txt"
+        )
+    ) as f:
         system += f.read()
 
     if history:

{wcgw-2.8.3/src/wcgw/client → wcgw-2.8.5/src/wcgw_cli}/cli.py

@@ -1,11 +1,11 @@
 import importlib
 from typing import Optional
-from typer import Typer
-import typer
 
-from .openai_client import loop as openai_loop
-from .anthropic_client import loop as claude_loop
+import typer
+from typer import Typer
 
+from wcgw_cli.anthropic_client import loop as claude_loop
+from wcgw_cli.openai_client import loop as openai_loop
 
 app = Typer(pretty_exceptions_show_locals=False)
 

{wcgw-2.8.3/src/wcgw/client → wcgw-2.8.5/src/wcgw_cli}/openai_client.py

@@ -23,7 +23,17 @@ from openai.types.chat import (
 from pydantic import BaseModel
 from typer import Typer
 
-from ..types_ import (
+from wcgw.client.common import CostData, History, Models, discard_input
+from wcgw.client.memory import load_memory
+from wcgw.client.tools import (
+    DoneFlag,
+    ImageData,
+    default_enc,
+    get_tool_output,
+    initialize,
+    which_tool,
+)
+from wcgw.types_ import (
     BashCommand,
     BashInteraction,
     ContextSave,
@@ -33,17 +43,8 @@ from ..types_ import (
     ResetShell,
     WriteIfEmpty,
 )
-from .common import CostData, History, Models, discard_input
-from .memory import load_memory
+
 from .openai_utils import get_input_cost, get_output_cost
-from .tools import (
-    DoneFlag,
-    ImageData,
-    default_enc,
-    get_tool_output,
-    initialize,
-    which_tool,
-)
 
 
 class Config(BaseModel):
@@ -235,7 +236,11 @@ Saves provided description and file contents of all the relevant file paths or g
         mode="wcgw",
     )
 
-    with open(os.path.join(os.path.dirname(__file__), "diff-instructions.txt")) as f:
+    with open(
+        os.path.join(
+            os.path.dirname(__file__), "..", "wcgw", "client", "diff-instructions.txt"
+        )
+    ) as f:
         system += f.read()
 
     if not history:

{wcgw-2.8.3/src/wcgw/client → wcgw-2.8.5/src/wcgw_cli}/openai_utils.py

@@ -1,25 +1,14 @@
-import json
-from pathlib import Path
-import select
-import sys
-import termios
-import traceback
-import tty
-from typing import Callable, DefaultDict, Literal, Optional, cast
-import openai
-from openai import OpenAI
+from typing import cast
+
 from openai.types.chat import (
-    ChatCompletionMessageParam,
     ChatCompletionAssistantMessageParam,
     ChatCompletionMessage,
+    ChatCompletionMessageParam,
     ParsedChatCompletionMessage,
 )
-import rich
 from tokenizers import Tokenizer  # type: ignore[import-untyped]
-from typer import Typer
-import uuid
 
-from .common import CostData, History
+from wcgw.client.common import CostData, History
 
 
 def get_input_cost(

{wcgw-2.8.3 → wcgw-2.8.5}/tests/client/test_anthropic_client_utils.py

@@ -1,15 +1,15 @@
 import json
-import os
-import pytest
-from pathlib import Path
 import tempfile
-from wcgw.client.anthropic_client import (
-    text_from_editor,
-    save_history,
+from unittest.mock import mock_open, patch
+
+import pytest
+import rich.console
+
+from wcgw_cli.anthropic_client import (
     parse_user_message_special,
+    save_history,
+    text_from_editor,
 )
-import rich.console
-from unittest.mock import patch, mock_open, MagicMock
 
 
 @pytest.fixture
@@ -18,20 +18,24 @@ def console():
 
 
 def test_text_from_editor_direct_input(console):
-    with patch('builtins.input', return_value="Test input"):
+    with patch("builtins.input", return_value="Test input"):
         result = text_from_editor(console)
         assert result == "Test input"
 
 
 def test_text_from_editor_with_editor():
-    with tempfile.NamedTemporaryFile(mode='w', suffix='.tmp') as tf:
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".tmp") as tf:
         tf.write("Test content from editor")
         tf.flush()
-        
-        with patch('builtins.input', return_value=""):
-            with patch('os.environ.get', return_value="cat"):  # Use 'cat' as editor for testing
-                with patch('subprocess.run') as mock_run:
-                    with patch('builtins.open', mock_open(read_data="Test content from editor")):
+
+        with patch("builtins.input", return_value=""):
+            with patch(
+                "os.environ.get", return_value="cat"
+            ):  # Use 'cat' as editor for testing
+                with patch("subprocess.run") as mock_run:
+                    with patch(
+                        "builtins.open", mock_open(read_data="Test content from editor")
+                    ):
                         console = rich.console.Console()
                         result = text_from_editor(console)
                         assert result == "Test content from editor"
@@ -43,28 +47,28 @@ def test_save_history(tmp_path):
     history = [
         {"role": "system", "content": "System message"},
         {"role": "user", "content": "Test message"},
-        {"role": "assistant", "content": "Test response"}
+        {"role": "assistant", "content": "Test response"},
     ]
     session_id = "test123"
-    
+
     # Create the .wcgw directory
-    wcgw_dir = tmp_path / '.wcgw'
+    wcgw_dir = tmp_path / ".wcgw"
     wcgw_dir.mkdir(parents=True, exist_ok=True)
-    
+
     # Patch Path to return our temp directory
-    with patch('pathlib.Path', return_value=wcgw_dir):
+    with patch("pathlib.Path", return_value=wcgw_dir):
         # Create a mock file context
-        with patch('builtins.open', mock_open()) as mock_file:
+        with patch("builtins.open", mock_open()) as mock_file:
             save_history(history, session_id)
-            
+
             # Verify the file was opened
             mock_file.assert_called_once()
-            
+
             # Get all the written content by joining all write calls
-            written_content = ''
+            written_content = ""
             for call_args in mock_file().write.call_args_list:
                 written_content += call_args[0][0]
-            
+
             # Verify the content written was valid JSON
             parsed_content = json.loads(written_content)
             assert len(parsed_content) == len(history)
@@ -86,11 +90,11 @@ def test_parse_user_message_special_with_image(tmp_path):
     image_path = tmp_path / "test.png"
     with open(image_path, "wb") as f:
         f.write(b"fake image data")
-    
+
     msg = f"%image {str(image_path)}\nSome text"
-    
-    with patch('base64.b64encode', return_value=b"fake_base64"):
-        with patch('mimetypes.guess_type', return_value=("image/png", None)):
+
+    with patch("base64.b64encode", return_value=b"fake_base64"):
+        with patch("mimetypes.guess_type", return_value=("image/png", None)):
             result = parse_user_message_special(msg)
             assert result["role"] == "user"
             assert isinstance(result["content"], list)
@@ -100,4 +104,4 @@ def test_parse_user_message_special_with_image(tmp_path):
             assert result["content"][0]["source"]["media_type"] == "image/png"
             assert result["content"][0]["source"]["data"] == "fake_base64"
             assert result["content"][1]["type"] == "text"
-            assert result["content"][1]["text"] == "Some text"
+            assert result["content"][1]["text"] == "Some text"

{wcgw-2.8.3 → wcgw-2.8.5}/tests/client/test_openai_utils.py

@@ -2,7 +2,7 @@ import pytest
 from typing import cast
 from openai.types.chat import ChatCompletionMessage, ChatCompletionMessageParam, ParsedChatCompletionMessage
 from tokenizers import Tokenizer
-from wcgw.client.openai_utils import get_input_cost, get_output_cost
+from wcgw_cli.openai_utils import get_input_cost, get_output_cost
 from wcgw.client.common import CostData
 
 

{wcgw-2.8.3 → wcgw-2.8.5}/tests/client/test_tools_extended.py

@@ -290,10 +290,9 @@ class TestToolsExtended(unittest.TestCase):
 
     @patch("wcgw.client.tools.command_run")
     def test_read_files_docker(self, mock_command_run):
-        from wcgw.client.tools import BASH_STATE, DisableConsole, read_files
+        from wcgw.client.tools import BASH_STATE, read_files
 
         # Setup mocks and test environment
-        console = DisableConsole()
         BASH_STATE.set_in_docker("test_container")
         mock_command_run.return_value = (0, "file content", "")
 

wcgw-2.8.5/tests/client/tools/test_full_coverage.py

@@ -0,0 +1,66 @@
+import logging
+import unittest
+from unittest.mock import MagicMock, patch
+
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+from wcgw.client.tools import (
+    get_incremental_output,
+    render_terminal_output,
+    truncate_if_over,
+)
+
+
+class TestFullCoverage(unittest.TestCase):
+    def setUp(self):
+        self.maxDiff = None
+
+    def test_incremental_output_single_case(self):
+        """Simple test for get_incremental_output function"""
+        old_output = ["line1", "line2"]
+        new_output = ["line2", "line3"]
+        result = get_incremental_output(old_output, new_output)
+        self.assertEqual(result, ["line3"])
+
+    def test_render_terminal_basic(self):
+        """Test render_terminal_output function"""
+        # Test basic output
+        result = render_terminal_output("Hello\nWorld")
+        self.assertEqual([line.rstrip() for line in result], ["Hello", "World"])
+
+        # Test with ANSI escape codes
+        result = render_terminal_output("\x1b[32mColored\x1b[0m\nText")
+        self.assertEqual([line.rstrip() for line in result], ["Colored", "Text"])
+
+    def test_truncate_if_over(self):
+        """Test truncate_if_over function"""
+        # Test with no truncation needed
+        with patch("wcgw.client.tools.default_enc") as mock_enc:
+            mock_enc.encode.return_value = [1, 2, 3]  # Under limit
+            result = truncate_if_over("short content", max_tokens=100)
+            self.assertEqual(result, "short content")
+
+        # Test with truncation needed
+        with patch("wcgw.client.tools.default_enc") as mock_enc:
+            mock_encoding = MagicMock()
+            mock_encoding.ids = list(range(200))  # Over limit
+            mock_encoding.__len__.return_value = 200
+            mock_enc.encode.return_value = mock_encoding
+            mock_enc.decode.return_value = "truncated"
+
+            result = truncate_if_over("long content", max_tokens=50)
+            self.assertIn("truncated", result)
+            self.assertIn("(...truncated)", result)
+
+        # Test with None max_tokens
+        result = truncate_if_over("any content", max_tokens=None)
+        self.assertEqual(result, "any content")
+
+        # Test with zero max_tokens
+        result = truncate_if_over("any content", max_tokens=0)
+        self.assertEqual(result, "any content")
+
+
+if __name__ == "__main__":
+    unittest.main()

wcgw-2.8.5/tests/client/tools/test_terminal_output_full.py

@@ -0,0 +1,114 @@
+"""Additional tests for terminal output handling in tools.py"""
+
+import unittest
+from unittest.mock import patch
+
+from wcgw.client.tools import (
+    Confirmation,
+    _incremental_text,
+    _is_int,
+    ask_confirmation,
+    get_incremental_output,
+    render_terminal_output,
+)
+
+
+class TestTerminalOutputFull(unittest.TestCase):
+    def setUp(self):
+        """Set up test environment before each test case"""
+        self.maxDiff = None
+
+    def test_get_incremental_output_break_conditions(self):
+        """Test break conditions in get_incremental_output"""
+        # Test case 1: No match found - returns entire new output
+        old_output = ["old"]
+        new_output = ["diff1", "diff2"]
+        result = get_incremental_output(old_output, new_output)
+        self.assertEqual(result, new_output)  # No match, return all
+
+        # Test case 2: Last line matches but not complete sequence - returns entire new output
+        old_output = ["start", "last"]
+        new_output = [
+            "other",
+            "last",
+            "extra",
+        ]  # Matches last line but not full sequence
+        result = get_incremental_output(old_output, new_output)
+        self.assertEqual(result, new_output)  # Incomplete match, return all
+
+        # Test case 3: Empty old output - returns entire new output
+        old_output = []
+        new_output = ["line1", "line2"]
+        result = get_incremental_output(old_output, new_output)
+        self.assertEqual(result, new_output)  # Empty old, return all
+
+        # Test case 4: Complete sequence match - returns only new content after match
+        old_output = ["start", "middle"]
+        new_output = ["start", "middle", "extra"]  # Complete sequence matches
+        result = get_incremental_output(old_output, new_output)
+        self.assertEqual(result, ["extra"])  # Complete match, return only new content
+
+    def test_incremental_terminal_output(self):
+        """Test incremental terminal output handling"""
+        # Test with completely empty old output
+        result = _incremental_text("new text", "")
+        self.assertEqual(result.rstrip(), "new text")
+
+        # Test with new output matching end of old
+        result = _incremental_text(
+            "some existing text\nnew text", "some existing text\n"
+        )
+        self.assertEqual(result.rstrip(), "new text")
+
+    def test_ask_confirmation(self):
+        """Test ask_confirmation function"""
+        with patch("builtins.input", return_value="y"):
+            result = ask_confirmation(Confirmation(prompt="Test prompt"))
+            self.assertEqual(result, "Yes")
+
+        with patch("builtins.input", return_value="n"):
+            result = ask_confirmation(Confirmation(prompt="Test prompt"))
+            self.assertEqual(result, "No")
+
+        # Test with other input that should be treated as no
+        with patch("builtins.input", return_value="anything"):
+            result = ask_confirmation(Confirmation(prompt="Test prompt"))
+            self.assertEqual(result, "No")
+
+    def test_terminal_empty_output(self):
+        """Test render_terminal_output with empty lines and whitespace"""
+        # Test with completely empty output
+        result = render_terminal_output("")
+        self.assertEqual(result, [])
+
+        # Test with non-empty followed by empty lines
+        result = render_terminal_output("line1\nline2\n  \n\n")
+        result = [line.rstrip() for line in result]  # Strip trailing spaces
+        self.assertEqual(result, ["line1", "line2"])  # Empty lines at end are filtered
+
+        # Test with content between empty lines
+        result = render_terminal_output("\n\nline1\n\nline2\n")
+        result = [line.rstrip() for line in result]
+        self.assertEqual(
+            result, ["", "", "line1", "", "line2"]
+        )  # Empty at start kept, at end filtered
+
+        # Test with trailing whitespace lines
+        result = render_terminal_output("line1\n   \n  ")
+        result = [line.rstrip() for line in result]
+        self.assertEqual(result, ["line1"])  # Whitespace at end filtered
+
+    def test_is_int_function(self):
+        """Test _is_int function with various inputs"""
+        self.assertTrue(_is_int("123"))  # Valid integer
+        self.assertTrue(_is_int("-123"))  # Negative integer
+        self.assertTrue(_is_int("0"))  # Zero
+
+        self.assertFalse(_is_int("abc"))  # Letters
+        self.assertFalse(_is_int("12.3"))  # Float
+        self.assertFalse(_is_int(""))  # Empty string
+        self.assertFalse(_is_int(" "))  # Whitespace
+
+
+if __name__ == "__main__":
+    unittest.main()