def test(self):
     test_data = [
         {
             'input': '[4,2,7,1,3]',
             'target': 2,
             'expected': '[2,1,3]'
         },
         {
             'input': '[4,2,7,1,3]',
             'target': 100,
             'expected': '[]'
         },
         {
             'input': '[8,3,10,1,6,null,14,null,null,4,7,13]',
             'target': 3,
             'expected': '[3,1,6,null,null,4,7]'
         },
         {
             'input': '[]',
             'target': 0,
             'expected': '[]'
         },
     ]
     for data in test_data:
         tree_str = data['input']
         target = data['target']
         expected = data['expected']
         with self.subTest(tree_str=tree_str,
                           target=target,
                           expected=expected):
             root = deserialize_tree_str(tree_str)
             output = self.solution.searchBST(root, target)
             self.assertEqual(serialize_treenode(output), expected)
 def test(self):
     test_data = [
         {'input': '[4,2,7,1,3,6,9]', 'expected': '[4,7,2,9,6,3,1]'},
         {'input': '[]', 'expected': '[]'},
     ]
     for data in test_data:
         tree_str = data['input']
         expected = data['expected']
         with self.subTest(tree_str=tree_str, expected=expected):
             root = deserialize_tree_str(tree_str)
             output = self.solution.invertTree(root)
             self.assertEqual(serialize_treenode(output), expected)
 def test(self):
     test_data = [
         {
             'input': '[]'
         },
         {
             'input': '[-1,0,1]'
         },
         {
             'input': '[1,2,3,4,null,2,4,null,null,4]'
         },
         {
             'input': '[1,2,3,null,4]'
         },
         {
             'input': '[1,2,3,null,null,4,5]'
         },
         {
             'input': '[1,null,2,3]'
         },
         {
             'input': '[1,null,2,null,3]'
         },
         {
             'input': '[2,1,3]'
         },
         {
             'input': '[3,1,null,null,2]'
         },
         {
             'input': '[3,2,null,1]'
         },
         {
             'input': '[3,9,20,null,null,15,7]'
         },
         {
             'input': '[4,2,7,1,3,6,9]'
         },
         {
             'input': '[5,4,7,3,null,2,null,-1,null,9]'
         },
         {
             'input': '[8,3,10,1,6,null,14,null,null,4,7,13]'
         },
     ]
     for data in test_data:
         tree_str = data['input']
         with self.subTest(tree_str=tree_str):
             root = deserialize_tree_str(tree_str)
             codec = PreorderCodec()
             expected_root = codec.deserialize(codec.serialize(root))
             self.assertEqual(tree_str, serialize_treenode(expected_root))
Exemplo n.º 4
0
 def test(self):
     test_data = [
         {'input': '[]'},
         {'input': '[-1,0,1]'},
         {'input': '[1,2,3,null,null,4,5]'},
     ]
     for data in test_data:
         tree_str = data['input']
         with self.subTest(tree_str=tree_str):
             root = deserialize_tree_str(tree_str)
             codec = Codec()
             expected_root = codec.deserialize(codec.serialize(root))
             self.assertEqual(tree_str, serialize_treenode(expected_root))