def test_values(self): """Tests the various non-default values.""" d1 = nested_dict([[1,2,3],["a","b","c"]]) d1[2]['b'].append("foo") ref_d1 = { 1:{"a":[], "b":[], "c":[]}, 2:{"a":[], "b":['foo',], "c":[]}, 3:{"a":[], "b":[], "c":[]}, } self.assertEquals(d1, ref_d1) d2 = nested_dict([[1,2,3],["a","b","c"]], value=False) d2[1]['a']=True d2[3]['c']=True ref_d2 = { 1:{"a":True, "b":False, "c":False}, 2:{"a":False, "b":False, "c":False}, 3:{"a":False, "b":False, "c":True}, } self.assertEquals(d2, ref_d2) d3 = nested_dict([[1,2,3],["a","b","c"]], value=0) d3[1]['a']=5 d3[3]['c']=3 ref_d3 = { 1:{"a":5, "b":0, "c":0}, 2:{"a":0, "b":0, "c":0}, 3:{"a":0, "b":0, "c":3}, } self.assertEquals(d3, ref_d3)
def test_basic(self): d1 = nested_dict([[1,2,3],["a","b","c"]]) ref_d1 = { 1:{"a":[], "b":[], "c":[]}, 2:{"a":[], "b":[], "c":[]}, 3:{"a":[], "b":[], "c":[]}, } self.assertEquals(d1, ref_d1) d2 = nested_dict([[1,2],'xy',[10,20]]) ref_d2 = { 1: { 'x': { 10: [], 20: [], }, 'y': { 10: [], 20: [], }, }, 2: { 'x': { 10: [], 20: [], }, 'y': { 10: [], 20: [], }, }, } self.assertEquals(d2, ref_d2)