File size: 1,841 Bytes
1678967 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
"""Tests for mathematical utility functions."""
import pytest
from src.utils.math import calculate_mean, calculate_median
def test_calculate_mean():
"""Test the calculate_mean function with various inputs."""
# Test with integers
assert calculate_mean([1, 2, 3, 4, 5]) == 3.0
# Test with floats
assert calculate_mean([1.5, 2.5, 3.5]) == 2.5
# Test with mixed numbers
assert calculate_mean([1, 2.5, 3]) == 2.1666666666666665
def test_calculate_mean_errors():
"""Test error handling in calculate_mean function."""
# Test empty list
with pytest.raises(ValueError) as exc:
calculate_mean([])
assert str(exc.value) == "Cannot calculate mean of empty list"
# Test invalid input types
with pytest.raises(TypeError) as exc:
calculate_mean([1, "2", 3])
assert str(exc.value) == "All elements must be numbers"
def test_calculate_median():
"""Test the calculate_median function with various inputs."""
# Test odd number of integers
assert calculate_median([1, 2, 3, 4, 5]) == 3.0
# Test even number of integers
assert calculate_median([1, 2, 3, 4]) == 2.5
# Test unsorted list
assert calculate_median([5, 2, 1, 4, 3]) == 3.0
# Test with floats
assert calculate_median([1.5, 2.5, 3.5]) == 2.5
# Test with mixed numbers
assert calculate_median([1, 2.5, 3]) == 2.5
def test_calculate_median_errors():
"""Test error handling in calculate_median function."""
# Test empty list
with pytest.raises(ValueError) as exc:
calculate_median([])
assert str(exc.value) == "Cannot calculate median of empty list"
# Test invalid input types
with pytest.raises(TypeError) as exc:
calculate_median([1, "2", 3])
assert str(exc.value) == "All elements must be numbers" |