def test_xdict_map_to_vals(self):
def cap_val(xdict_val):
return xdict_val.upper()
xd = XDict({'a_one': 'value_one', 'b_two': 'value_two', 'a_three': 'value_three'}, {'attrkey': 'attrval'})
xd = xd.map_to_vals(cap_val)
self.assertEqual(type(xd), type(XDict({'test': 'dict'})))
self.assertTrue(len(xd.values()) == 3)
self.assertTrue('VALUE_ONE' in xd.values())
self.assertTrue('VALUE_TWO' in xd.values())
self.assertTrue('VALUE_THREE' in xd.values())
self.assertEqual(xd.attrkey, 'attrval')
def test_xdict_conditional_map_to_vals(self):
def true_a(xdict_key):
return xdict_key.startswith('a')
def cap_val(xdict_val):
return xdict_val.upper()
xd = XDict({'a_one': 'value_one', 'b_two': 'value_two', 'a_three': 'value_three'}, {'attrkey': 'attrval'})
xd = xd.conditional_map_to_vals(true_a, cap_val)
self.assertTrue(len(xd.values()) == 3)
self.assertTrue('VALUE_ONE' in xd.values())
self.assertTrue('VALUE_THREE' in xd.values())
self.assertTrue('value_two' in xd.values())
self.assertFalse('VALUE_TWO' in xd.values()) # the b_two value was not converted as per the conditional function
def test_xdict_map_to_vals(self):
def cap_val(xdict_val):
return xdict_val.upper()
xd = XDict(
{
'a_one': 'value_one',
'b_two': 'value_two',
'a_three': 'value_three'
}, {'attrkey': 'attrval'})
xd = xd.map_to_vals(cap_val)
self.assertEqual(type(xd), type(XDict({'test': 'dict'})))
self.assertTrue(len(xd.values()) == 3)
self.assertTrue('VALUE_ONE' in xd.values())
self.assertTrue('VALUE_TWO' in xd.values())
self.assertTrue('VALUE_THREE' in xd.values())
self.assertEqual(xd.attrkey, 'attrval')
def test_xdict_conditional_map_to_vals(self):
def true_a(xdict_key):
return xdict_key.startswith('a')
def cap_val(xdict_val):
return xdict_val.upper()
xd = XDict(
{
'a_one': 'value_one',
'b_two': 'value_two',
'a_three': 'value_three'
}, {'attrkey': 'attrval'})
xd = xd.conditional_map_to_vals(true_a, cap_val)
self.assertTrue(len(xd.values()) == 3)
self.assertTrue('VALUE_ONE' in xd.values())
self.assertTrue('VALUE_THREE' in xd.values())
self.assertTrue('value_two' in xd.values())
self.assertFalse('VALUE_TWO' in xd.values(
)) # the b_two value was not converted as per the conditional function
