python/ヤコビ法の収束回数が変わらない

前提

pythonでヤコビ法のプログラムを作りましたが、値を変えても収束回数が全て2になります。どこが間違っていますか？

プログラムの条件は
b = Axの方程式に対し、対角優位な5×5の行列Aおよび入力ベクトルbを作成し（値はテキトー）、反復法を用いて解x を求める。計算したベクトルのノルムの差が1.0e-7以下であれば、収束したものとする（収束するまでかかった回数を画面に出力する）。
です。

該当のソースコード

python
1A = np.array([[1, 0, 0, 0, 0], [0, 1, 0, 0, 0], [0, 0, 1, 0, 0], [0, 0, 0, 1, 0], [0, 0, 0, 0, 1]])
2b = [1, 2, 3, 4, 5]
3x_o = [0, 0, 0, 0, 0] #初期値
4count = 0
5
6a = np. diag(A) #Aの対角成分ベクトル
7D = np. diagflat(a) #aベクトルを対角成分に持つ行列
8D_inv = np.linalg.inv(D) #Dの逆行列
9
10print(A)
11
12for i in range(1000):
13  x_n =  D_inv @ (b - (A - D) @ x_o)
14  err = np.linalg.norm(x_n - x_o)
15  count += 1
16  x_o = x_n
17  if err < 1.0e-7:
18    break
19
20print("収束するまでかかった回数:", count)
21