mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-04-27 16:57:34 -05:00
0a53597e27
✅ Setup Module Implementation: - Created comprehensive setup_dev.ipynb with TinyTorch workflow tutorial - Added hello_tinytorch(), add_numbers(), and SystemInfo class - Updated README with clear learning objectives and development workflow - All 11 tests passing for complete workflow validation 🔧 CLI Enhancements: - Added --module flag to 'tito sync' for module-specific exports - Implemented 'tito reset' command with --force option - Smart auto-generated file detection and cleanup - Interactive confirmation with safety preservations 📚 Documentation Updates: - Updated all references to use [module]_dev.ipynb naming convention - Enhanced test coverage for new functionality - Clear error handling and user guidance This establishes the foundation workflow that students will use throughout TinyTorch development.
86 lines
3.0 KiB
Python
86 lines
3.0 KiB
Python
"""
|
|
Tests for Module 0: Setup & Environment
|
|
|
|
These tests validate the exported functions from modules/setup/setup_dev.ipynb
|
|
"""
|
|
|
|
import pytest
|
|
from tinytorch.core.utils import hello_tinytorch, add_numbers, SystemInfo
|
|
|
|
|
|
def test_hello_tinytorch_exists():
|
|
"""Test that the hello_tinytorch function exists and can be imported."""
|
|
assert callable(hello_tinytorch), "hello_tinytorch should be a callable function"
|
|
|
|
|
|
def test_hello_tinytorch_returns_string():
|
|
"""Test that hello_tinytorch returns a string."""
|
|
result = hello_tinytorch()
|
|
assert isinstance(result, str), f"hello_tinytorch() should return a string, got {type(result)}"
|
|
|
|
|
|
def test_hello_tinytorch_not_empty():
|
|
"""Test that hello_tinytorch returns a non-empty string."""
|
|
result = hello_tinytorch()
|
|
assert len(result.strip()) > 0, "hello_tinytorch() should return a non-empty string"
|
|
|
|
|
|
def test_hello_tinytorch_contains_tinytorch():
|
|
"""Test that the greeting contains TinyTorch."""
|
|
result = hello_tinytorch().lower()
|
|
assert 'tinytorch' in result, f"hello_tinytorch() should contain 'TinyTorch'. Got: {hello_tinytorch()}"
|
|
|
|
|
|
def test_hello_tinytorch_has_fire_emoji():
|
|
"""Test that the greeting contains the fire emoji (matching brand)."""
|
|
result = hello_tinytorch()
|
|
assert '🔥' in result, f"hello_tinytorch() should contain the 🔥 emoji to match TinyTorch branding. Got: {result}"
|
|
|
|
|
|
def test_add_numbers_basic():
|
|
"""Test the add_numbers function with basic inputs."""
|
|
assert add_numbers(2, 3) == 5
|
|
assert add_numbers(0, 0) == 0
|
|
assert add_numbers(-1, 1) == 0
|
|
assert add_numbers(10, -5) == 5
|
|
|
|
|
|
def test_add_numbers_float():
|
|
"""Test add_numbers with floating point numbers."""
|
|
assert abs(add_numbers(1.5, 2.5) - 4.0) < 1e-10
|
|
assert abs(add_numbers(0.1, 0.2) - 0.3) < 1e-10
|
|
|
|
|
|
def test_system_info_creation():
|
|
"""Test that SystemInfo can be created."""
|
|
info = SystemInfo()
|
|
assert info is not None
|
|
assert hasattr(info, 'python_version')
|
|
assert hasattr(info, 'platform')
|
|
assert hasattr(info, 'machine')
|
|
|
|
|
|
def test_system_info_string_representation():
|
|
"""Test that SystemInfo has a proper string representation."""
|
|
info = SystemInfo()
|
|
info_str = str(info)
|
|
assert isinstance(info_str, str)
|
|
assert len(info_str) > 0
|
|
assert 'Python' in info_str
|
|
|
|
|
|
def test_system_info_compatibility():
|
|
"""Test the is_compatible method."""
|
|
info = SystemInfo()
|
|
# Should be compatible since we're running this test
|
|
assert info.is_compatible() == True
|
|
|
|
|
|
def test_function_docstrings():
|
|
"""Test that exported functions have proper docstrings."""
|
|
assert hello_tinytorch.__doc__ is not None, "hello_tinytorch should have a docstring"
|
|
assert add_numbers.__doc__ is not None, "add_numbers should have a docstring"
|
|
|
|
# Check that docstrings are meaningful (not just empty)
|
|
assert len(hello_tinytorch.__doc__.strip()) > 5, "Docstring should be meaningful"
|
|
assert len(add_numbers.__doc__.strip()) > 5, "Docstring should be meaningful" |