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))
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))