# retoor <retoor@molodetz.nl>
import math
import unittest
from src.calculator import clamp, median
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 TestMedianFunction(unittest.TestCase):
def test_odd_length_returns_middle_element(self) -> None:
self.assertEqual(median([1, 3, 5]), 3)
def test_even_length_returns_float_average_of_two_middle_values(self) -> None:
result = median([1, 2, 3, 4])
self.assertIsInstance(result, float)
self.assertEqual(result, 2.5)
def test_single_element_returns_that_element(self) -> None:
self.assertEqual(median([7]), 7)
def test_empty_list_raises_value_error(self) -> None:
with self.assertRaises(ValueError):
median([])
def test_unsorted_input_sorts_correctly(self) -> None:
self.assertEqual(median([3, 1, 2]), 2)
def test_unsorted_even_length_returns_float_average(self) -> None:
result = median([10, 30, 20, 40])
self.assertIsInstance(result, float)
self.assertEqual(result, 25.0)