def METHOD_NAME(self):
    root = source_suite_tree(dedent("""\
        if True:
            pass
        print(1)
        if True:
            pass
    """))
    if_suite1 = root.get_children()[0]
    if_suite2 = root.get_children()[1]
    self.assertEqual(if_suite1, root.find_suite(2))
    self.assertEqual(if_suite2, root.find_suite(5))
    self.assertEqual(root, root.find_suite(3))