def METHOD_NAME():
    mesh1 = discretize.TreeMesh([64, 64])
    mesh1.refine_ball([0.5, 0.5], [0.1], -1)
    bad_nodes = check_for_diag_unbalance(mesh1)
    assert len(bad_nodes) == 12
    mesh2 = discretize.TreeMesh([64, 64], diagonal_balance=True)
    mesh2.refine_ball([0.5, 0.5], [0.1], -1)
    bad_nodes = check_for_diag_unbalance(mesh2)
    assert len(bad_nodes) == 0