# Path: nba_data/data/unit.py
# class Unit(Enum):
#     starters = "starters"
#     bench = "bench"
# 
#     @staticmethod
#     def get_unit_from_name(unit_name):
#         unit = unit_name_to_unit_map.get(unit_name.upper())
# 
#         if unit is None:
#             raise ValueError("Unknown unit name: %s", unit_name)
# 
#         return unit


from unittest import TestCase
from nba_data.data.unit import Unit



class TestUnit(TestCase):
    def test_get_unit_from_name(self):
        self.assertRaises(ValueError, Unit.get_unit_from_name, "bae jadley")

# Path: nba_data/data/unit.py
# class Unit(Enum):
#     starters = "starters"
#     bench = "bench"
# 
#     @staticmethod
#     def get_unit_from_name(unit_name):
#         unit = unit_name_to_unit_map.get(unit_name.upper())
# 
#         if unit is None:
#             raise ValueError("Unknown unit name: %s", unit_name)
# 
#         return unit


# Path: tests/test_unit.py
from unittest import TestCase
from nba_data.data.unit import Unit




class TestUnit(TestCase):
    def test_get_unit_from_name(self):
        self.assertRaises(ValueError, Unit.get_unit_from_name, "bae jadley")

        self.assertEqual(Unit.bench, Unit.get_unit_from_name("bEnCh"))