def test_add_child(self): node = TreeNode() child = mock.MagicMock() ret = node.add_child(child) self.assertIs(child, ret) self.assertIs(node.child, child)
def test_is_expect(self): node = TreeNode() with self.assertRaises(NotImplementedError): node.is_expect()
def test_is_generator(self): node = TreeNode() with self.assertRaises(NotImplementedError): node.is_generator()
def test_is_command(self): node = TreeNode() with self.assertRaises(NotImplementedError): node.is_command()
def test_get_all_siblings_with_siblings(self): node = TreeNode() sibling = TreeNode() node.next_sibling = sibling self.assertEqual([node, sibling], list(node.get_all_siblings()))
def test_get_all_siblings(self): node = TreeNode() self.assertEqual([node], list(node.get_all_siblings()))
def test___init__(self): node = TreeNode() self.assertIsNotNone(node) self.assertIsNone(node.child) self.assertIsNone(node.next_sibling)