def __init__(
self,
expanded_access_type_individual=None,
expanded_access_type_intermediate=None,
expanded_access_type_treatment=None,
):
self.individual = yes_no_enum(expanded_access_type_individual)
self.intermediate = yes_no_enum(expanded_access_type_intermediate)
self.treatment = yes_no_enum(expanded_access_type_treatment)
def has_expanded_access(self):
return yes_no_enum(
glom(self._data, "has_expanded_access", default="No"))
def test_wacky(self):
with self.assertRaises(ValueError) as exc:
yes_no_enum(1)
self.assertEqual(str(exc.exception),
"Unable to process value of type <class 'int'>")
def test_none(self):
self.assertFalse(yes_no_enum(None))
def test_no(self):
self.assertFalse(yes_no_enum("No"))
def test_yes(self):
self.assertTrue(yes_no_enum("Yes"))