From 8a371c786fc77139be2a325c05e0a5c95ea83246 Mon Sep 17 00:00:00 2001 From: kafkaf- Date: Fri, 14 Nov 2014 16:06:20 +0200 Subject: Removed the setUpClass method to support py26 --- tests.py | 80 +++++++++++++++++++++++++++++++--------------------------------- 1 file changed, 39 insertions(+), 41 deletions(-) diff --git a/tests.py b/tests.py index cc81dc6..3bec626 100755 --- a/tests.py +++ b/tests.py @@ -185,47 +185,45 @@ class SetTests(unittest.TestCase): class AltTypesTests(unittest.TestCase): - @classmethod - def setUpClass(cls): - class Node(object): - def __init__(self, name, parent=None): - self.name = name - self.parent = parent - self.left = None - self.right = None - - def set_left(self, node): - node.parent = self - self.left = node - - def set_right(self, node): - node.parent = self - self.right = node - - def __getitem__(self, key): - if key == 'left': - return self.left - if key == 'right': - return self.right - - raise KeyError("Only left and right supported") - - def __setitem__(self, key, val): - if key == 'left': - return self.set_left(val) - if key == 'right': - return self.set_right(val) - - raise KeyError("Only left and right supported: %s" % key) - - class mdict(object): - def __init__(self, d): - self._d = d - def __getitem__(self, item): - return self._d[item] - - cls.mdict = mdict({'root': {'1': {'2': '3'}}}) - cls.Node = Node + class Node(object): + def __init__(self, name, parent=None): + self.name = name + self.parent = parent + self.left = None + self.right = None + + def set_left(self, node): + node.parent = self + self.left = node + + def set_right(self, node): + node.parent = self + self.right = node + + def __getitem__(self, key): + if key == 'left': + return self.left + if key == 'right': + return self.right + + raise KeyError("Only left and right supported") + + def __setitem__(self, key, val): + if key == 'left': + return self.set_left(val) + if key == 'right': + return self.set_right(val) + + raise KeyError("Only left and right supported: %s" % key) + + class mdict(object): + def __init__(self, d): + self._d = d + def __getitem__(self, item): + return self._d[item] + + mdict = mdict({'root': {'1': {'2': '3'}}}) + Node = Node def test_alttypes(self): -- cgit v1.2.1