def test_extend(self): @Node.extend() def add(self, lhs, rhs): return lhs + rhs n = Node() self.assertEqual(n.add(4, 2), 6) delattr(Node, 'add') self.assertRaises(AttributeError, lambda: n.add(2, 4))