Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2f86b1c6cc | ||
|
|
3c5079a9d8 |
+8
-7
@@ -1,5 +1,8 @@
|
||||
# retoor <retoor@molodetz.nl>
|
||||
|
||||
from typing import Sequence
|
||||
|
||||
|
||||
def add(left: int, right: int) -> int:
|
||||
return left + right
|
||||
|
||||
@@ -22,12 +25,10 @@ def clamp_to_byte(value: int) -> int:
|
||||
return max(0, min(255, value))
|
||||
|
||||
|
||||
def median(values: list[float]) -> float:
|
||||
def variance(values: Sequence[float]) -> float:
|
||||
if not values:
|
||||
raise ValueError
|
||||
sorted_values = sorted(values)
|
||||
n = len(sorted_values)
|
||||
mid = n // 2
|
||||
if n % 2 == 1:
|
||||
return sorted_values[mid]
|
||||
return (sorted_values[mid - 1] + sorted_values[mid]) / 2.0
|
||||
mean = sum(values) / len(values)
|
||||
return sum((x - mean) ** 2 for x in values) / len(values)
|
||||
|
||||
|
||||
|
||||
+26
-19
@@ -3,7 +3,7 @@
|
||||
import math
|
||||
import unittest
|
||||
|
||||
from src.calculator import clamp, median
|
||||
from src.calculator import clamp, variance
|
||||
|
||||
|
||||
class TestClampFunction(unittest.TestCase):
|
||||
@@ -63,27 +63,34 @@ class TestClampFunction(unittest.TestCase):
|
||||
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)
|
||||
class TestVarianceFunction(unittest.TestCase):
|
||||
|
||||
def test_empty_list_raises_value_error(self) -> None:
|
||||
with self.assertRaises(ValueError):
|
||||
median([])
|
||||
variance([])
|
||||
|
||||
def test_unsorted_input_sorts_correctly(self) -> None:
|
||||
self.assertEqual(median([3, 1, 2]), 2)
|
||||
def test_empty_tuple_raises_value_error(self) -> None:
|
||||
with self.assertRaises(ValueError):
|
||||
variance(())
|
||||
|
||||
def test_unsorted_even_length_returns_float_average(self) -> None:
|
||||
result = median([10, 30, 20, 40])
|
||||
def test_single_element_returns_zero(self) -> None:
|
||||
self.assertEqual(variance([42.0]), 0.0)
|
||||
|
||||
def test_constant_values_return_zero_variance(self) -> None:
|
||||
self.assertEqual(variance([1.0, 1.0, 1.0]), 0.0)
|
||||
|
||||
def test_population_variance_of_known_set(self) -> None:
|
||||
self.assertEqual(variance([1, 2, 3, 4, 5]), 2.0)
|
||||
|
||||
def test_two_element_variance(self) -> None:
|
||||
self.assertEqual(variance([0, 2]), 1.0)
|
||||
|
||||
def test_integer_inputs_return_float(self) -> None:
|
||||
result = variance([10, 20, 30])
|
||||
self.assertIsInstance(result, float)
|
||||
self.assertEqual(result, 25.0)
|
||||
self.assertEqual(result, 200.0 / 3.0)
|
||||
|
||||
def test_tuple_input_returns_variance(self) -> None:
|
||||
self.assertEqual(variance((1, 2, 3, 4, 5)), 2.0)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user