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