def test_to_dict_recursive_circular():
class Brother(TInterface):
_dict_attrs = ('name', 'siblings')
def __init__(self, name):
self.name = name
self.siblings = []
def __repr__(self):
return 'Brother: %s' % (self.name)
tom = Brother('tom')
jerry = Brother('jerry')
tom.siblings.append(jerry)
jerry.siblings.append(tom)
result = to_dict_recursive(tom)
result2 = to_dict_recursive(jerry)
assert_equal(result, {
'name': 'tom',
'siblings': [{
'name': 'jerry',
'siblings': []
}]
})
assert_equal(result2, {
'name': 'jerry',
'siblings': [{
'name': 'tom',
'siblings': []
}]
})
def test_to_dict_recursive_circular():
class Brother(TInterface):
_dict_attrs = (
'name',
'siblings'
)
def __init__(self, name):
self.name = name
self.siblings = []
def __repr__(self):
return 'Brother: %s' % (self.name)
tom = Brother('tom')
jerry = Brother('jerry')
tom.siblings.append(jerry)
jerry.siblings.append(tom)
result = to_dict_recursive(tom)
result2 = to_dict_recursive(jerry)
assert_equal(
result,
{'name': 'tom', 'siblings': [{'name': 'jerry', 'siblings': []}]}
)
assert_equal(
result2,
{'name': 'jerry', 'siblings': [{'name': 'tom', 'siblings': []}]}
)
def test_to_dict_recursive_multi_branch():
class Roots(TInterface):
_dict_attrs = ('trees', )
def __init__(self):
self.trees = [AppleTree(self), AppleTree(self), AppleTree(self)]
class AppleTree(TInterface):
_dict_attrs = ('root', 'apples', 'water', 'leaf_count')
def __init__(self, root):
self.root = root
self.apples = [Apple(self), Apple(self)]
self.water = choice(waters)
self.leaf_count = randint(0, 10)
class Apple(TInterface):
_dict_attrs = ('false_property', 'mah_property', 'tree', 'seeds',
'water', 'size')
@property
def mah_property(self):
return 'so proper!'
@property
def false_property(self):
return False
def __init__(self, tree):
self.tree = tree
self.seeds = [AppleSeed(self) for x in xrange(randint(1, 5))]
self.water = choice(waters)
self.size = choice(['small', 'medium', 'large'])
class AppleSeed(TInterface):
_dict_attrs = ('apple', 'water', 'potency')
def __init__(self, apple):
self.apple = apple
self.water = choice(waters)
self.potency = choice(['dull', 'potent', 'juicy'])
class Water(TInterface):
_dict_attrs = ('name', )
def __init__(self, name):
self.name = name
waters = [Water('ground'), Water('rain'), Water('bottle')]
result = to_dict_recursive(Roots())
def test_to_dict_recursive_multi_branch():
class Roots(TInterface):
_dict_attrs = (
'trees',
)
def __init__(self):
self.trees = [
AppleTree(self),
AppleTree(self),
AppleTree(self)
]
class AppleTree(TInterface):
_dict_attrs = (
'root',
'apples',
'water',
'leaf_count'
)
def __init__(self, root):
self.root = root
self.apples = [
Apple(self),
Apple(self)
]
self.water = choice(waters)
self.leaf_count = randint(0, 10)
class Apple(TInterface):
_dict_attrs = (
'false_property',
'mah_property',
'tree',
'seeds',
'water',
'size'
)
@property
def mah_property(self):
return 'so proper!'
@property
def false_property(self):
return False
def __init__(self, tree):
self.tree = tree
self.seeds = [AppleSeed(self) for x in xrange(randint(1, 5))]
self.water = choice(waters)
self.size = choice(['small', 'medium', 'large'])
class AppleSeed(TInterface):
_dict_attrs = (
'apple',
'water',
'potency'
)
def __init__(self, apple):
self.apple = apple
self.water = choice(waters)
self.potency = choice(['dull', 'potent', 'juicy'])
class Water(TInterface):
_dict_attrs = ('name',)
def __init__(self, name):
self.name = name
waters = [
Water('ground'),
Water('rain'),
Water('bottle')
]
result = to_dict_recursive(Roots())
