test_utils.py

import unittest
from src.utils import calculate_average, get_user_name


class UtilsTests(unittest.TestCase):
    def test_average_basic(self):
        self.assertEqual(calculate_average([2, 4, 6]), 4)

    def test_average_empty_returns_zero(self):
        self.assertEqual(calculate_average([]), 0.0)

    def test_user_name_upper(self):
        self.assertEqual(get_user_name({"name": "ada"}), "ADA")

    def test_user_name_none_returns_empty(self):
        self.assertEqual(get_user_name(None), "")

    def test_user_name_missing_key_returns_empty(self):
        self.assertEqual(get_user_name({}), "")


if __name__ == "__main__":
    unittest.main()