typosaurus-sandbox/tests/test_calculator.py

# retoor <retoor@molodetz.nl>

import math
import unittest

from src.calculator import average, clamp


class TestClampFunction(unittest.TestCase):

    def test_value_below_low_returns_low(self) -> None:
        self.assertEqual(clamp(-5, 0, 10), 0)

    def test_value_above_high_returns_high(self) -> None:
        self.assertEqual(clamp(15, 0, 10), 10)

    def test_value_in_range_returns_value(self) -> None:
        self.assertEqual(clamp(5, 0, 10), 5)

    def test_value_equals_low_returns_low(self) -> None:
        self.assertEqual(clamp(0, 0, 10), 0)

    def test_value_equals_high_returns_high(self) -> None:
        self.assertEqual(clamp(10, 0, 10), 10)

    def test_low_greater_than_high_raises_value_error(self) -> None:
        with self.assertRaises(ValueError):
            clamp(5, 10, 0)

    def test_all_negative_values(self) -> None:
        self.assertEqual(clamp(-10, -5, -1), -5)

    def test_float_value_below_low_returns_low_as_int(self) -> None:
        self.assertEqual(clamp(-1.0, 0, 10), 0)

    def test_float_value_above_high_returns_high_as_int(self) -> None:
        self.assertEqual(clamp(15.0, 0, 10), 10)

    def test_float_value_in_range_returns_float(self) -> None:
        result = clamp(5.0, 0, 10)
        self.assertIsInstance(result, float)
        self.assertEqual(result, 5.0)

    def test_float_value_equals_boundary_returns_boundary(self) -> None:
        self.assertEqual(clamp(0.0, 0, 10), 0)
        self.assertEqual(clamp(10.0, 0, 10), 10)

    def test_float_low_greater_than_float_high_raises_value_error(self) -> None:
        with self.assertRaises(ValueError):
            clamp(5.0, 10.0, 0.0)

    def test_large_values(self) -> None:
        self.assertEqual(clamp(10**9, 0, 10**6), 10**6)

    def test_negative_infinity_not_clamped_by_default(self) -> None:
        self.assertEqual(clamp(-math.inf, 0, 10), 0)

    def test_positive_infinity_not_clamped_by_default(self) -> None:
        self.assertEqual(clamp(math.inf, 0, 10), 10)

    def test_nan_returns_nan(self) -> None:
        result = clamp(math.nan, 0, 10)
        self.assertTrue(math.isnan(result))


class TestAverageFunction(unittest.TestCase):

    def test_empty_sequence_raises_value_error(self) -> None:
        with self.assertRaises(ValueError):
            average([])

    def test_single_element(self) -> None:
        self.assertEqual(average([5]), 5.0)

    def test_positive_values(self) -> None:
        self.assertEqual(average([1, 2, 3, 4, 5]), 3.0)

    def test_negative_values(self) -> None:
        self.assertEqual(average([-10, -20, -30]), -20.0)

    def test_mixed_positive_and_negative(self) -> None:
        self.assertEqual(average([-5, 0, 5]), 0.0)

    def test_float_values(self) -> None:
        self.assertEqual(average([1.5, 2.5, 3.0]), 7.0 / 3.0)