def test_problem2(self):
person_a = Person('User', '1', None)
person_b = Person('User', '2', person_a)
dataDict = {
'key1': 1,
'key2': {
'key3': 1,
'key4': {
'key5': 4,
'user': person_b
}
}
}
objects = {}
output = {
'key1': 1,
'key2': 1,
'key3': 2,
'key4': 2,
'key5': 3,
'user': 3,
'first_name': 4,
'last_name': 4,
'father': 4,
'first_name': 5,
'last_name': 5,
'father': 5
}
print_depth(dataDict, objects)
self.assertDictEqual(objects, output)
def test_depth_one(self):
a = {"key1": 1}
self.assertEqual(print_depth(a), 1)
def test_invalid_data_with_string(self):
with self.assertRaises(ValueError):
print_depth("")
def test_depth_two(self):
a = {"key1": {"key2": 2}}
self.assertEqual(print_depth(a), 2)
def test_empty(self):
self.assertAlmostEqual(print_depth({}), None)