RobotDaily/templates/test_template.py

"""
Day {{DAY_N}}: {{TOPIC_NAME}} - Test Suite

Use pytest to verify your implementation.
Run with: pytest tests/test_day{{DAY_N}}.py -v

Expected assertions:
- Tensor shapes match expectation
- Numerical values within tolerance
- Edge cases handled correctly
"""

import pytest
import numpy as np
import torch
from your_module import {{function_name}}  # Import your solution here


class TestDay{{DAY_N}}{{TOPIC_NAME}}:
    """Test suite for Day {{DAY_N}}: {{TOPIC_NAME}}"""

    def test_basic_case(self):
        """Test with basic, typical inputs"""
        # TODO: Add typical test case
        pytest.skip("Pending implementation")

    def test_shapes_match(self):
        """Verify output tensor shape matches expected"""
        # TODO: Add shape assertion
        pytest.skip("Pending implementation")

    def test_numerical_precision(self):
        """Use np.testing.assert_allclose for numerical comparison"""
        # TODO: Add precision assertion
        pytest.skip("Pending implementation")

    def test_edge_case_1(self):
        """Test edge case 1"""
        pytest.skip("Pending implementation")

    def test_edge_case_2(self):
        """Test edge case 2"""
        pytest.skip("Pending implementation")


class TestVisualizer:
    """Test the plot_concept() visualization"""

    def test_plot_renders(self):
        """Verify the plot function renders without error"""
        pytest.skip("Manual visual inspection required")


if __name__ == "__main__":
    pytest.main([__file__, "-v"])