def test_two_keys__good__with_default(self): func = __unit__.key_func('foo', 'bar', default=self.DEFAULT) self.assertEquals( self.DOUBLY_NESTED_DICT['foo']['bar'], func(self.DOUBLY_NESTED_DICT))
def test_two_keys__bad__with_default(self): func = __unit__.key_func('foo', 'doesnt_exist', default=self.DEFAULT) self.assertEquals(self.DEFAULT, func(self.DOUBLY_NESTED_DICT))
def test_single_key__good__with_default(self): func = __unit__.key_func('foo', default=self.DEFAULT) self.assertEquals( self.SINGLY_NESTED_DICT['foo'], func(self.SINGLY_NESTED_DICT)) self.assertEquals( self.DOUBLY_NESTED_DICT['foo'], func(self.DOUBLY_NESTED_DICT))
def test_single_key__bad__with_default(self): func = __unit__.key_func('doesnt_exist', default=self.DEFAULT) self.assertEquals(self.DEFAULT, func(self.SINGLY_NESTED_DICT)) self.assertEquals(self.DEFAULT, func(self.DOUBLY_NESTED_DICT))
def test_two_keys__good(self): func = __unit__.key_func('foo', 'bar') self.assertEquals( self.DOUBLY_NESTED_DICT['foo']['bar'], func(self.DOUBLY_NESTED_DICT))
def test_two_keys__bad(self): func = __unit__.key_func('doesnt_exist', 'foo') with self.assertRaises(LookupError): func(self.DOUBLY_NESTED_DICT)
def test_single_key__bad(self): func = __unit__.key_func('doesnt_exist') with self.assertRaises(LookupError): func(self.SINGLY_NESTED_DICT) with self.assertRaises(LookupError): func(self.DOUBLY_NESTED_DICT)
def test_single_key__good(self): func = __unit__.key_func('foo') self.assertEquals( self.SINGLY_NESTED_DICT['foo'], func(self.SINGLY_NESTED_DICT)) self.assertEquals( self.DOUBLY_NESTED_DICT['foo'], func(self.DOUBLY_NESTED_DICT))
def test_some_object(self): with self.assertRaises(TypeError): __unit__.key_func(object())
def test_none(self): with self.assertRaises(TypeError): __unit__.key_func(None)
def test_no_args(self): with self.assertRaises(TypeError): __unit__.key_func()