def METHOD_NAME():
    mesh1 = discretize.TreeMesh([64, 64])
    mesh1.insert_cells([0.09, 0.09], -1, finalize=False)
    mesh1.insert_cells([0.09, 0.91], -1, finalize=False)
    mesh1.insert_cells([0.91, 0.09], -1, finalize=False)
    mesh1.insert_cells([0.91, 0.91], -1)
    bad_nodes = check_for_diag_unbalance(mesh1)
    assert len(bad_nodes) == 8
    mesh2 = discretize.TreeMesh([64, 64], diagonal_balance=True)
    mesh2.insert_cells([0.09, 0.09], -1, finalize=False)
    mesh2.insert_cells([0.09, 0.91], -1, finalize=False)
    mesh2.insert_cells([0.91, 0.09], -1, finalize=False)
    mesh2.insert_cells([0.91, 0.91], -1)
    bad_nodes = check_for_diag_unbalance(mesh2)
    assert len(bad_nodes) == 0