[Python]関数デコレータを用いてゼロ除算が行われていないかを調べ、行われている場合はプログラムを中止したい

Question

タイトルを実現するために、コード欄のようなプログラムを以下の考え方で作成しました。
・A,B2つの行列を作る
・演算子のオーバーロードを用いて行列の要素ごとの割り算を行う
・関数デコレータを用いてゼロ除算がないかどうか調べる
これを実行すると以下のような結果が得られました。
ーーーーーーーーーーーーーーーーーーーーーーー
C=A/B:
C:OOOOOOOOOOOOOOOOOOOOOO: RuntimeWarning: divide by zero encountered in true_divide
print(A/B)
[[ inf 0.56017643]
[0.36731304 0.47638696]]
ーーーーーーーーーーーーーーーーーーーーーーー
このような結果が得られるということは
関数デコレータが実行されておらず、割り算を行う関数のみが働いているということなのでしょうか。
デコレータ内のif文の両方の条件分岐にprint文を記述しても
実行結果は変わらなかったため、関数デコレータは実行されていないと思います。
どのような理由で実行されていないか教えていただけるとありがたいです。

import sys
import os
import numpy as np
def zero(func):
	def inner(A, B):
		if np.all(B != 0):
			return func(a, b)
		else:
			print('ゼロ除算が発生します')
			sys.exit()
			return
	return inner
class Matrix:
	def __init__(self, M, N, mat):
		self.__M = M
		self.__N = N
		self.__mat = mat
	@zero
	def __truediv__(self, another):	
		return np.divide(self.__mat,another)
if __name__ == "__main__":
	A = np.random.random((2,2))
	B = np.random.random((2,2))
	B[0,0] = 0.0
	C = Matrix(2, 2, A)
	print("C=A/B:")
	print(A/B)

Comments

[quickquip]

> これを実行すると以下のような結果が得られました。
の内容とコードが一致していないように感じられるのですが。

Answer 1

@zero デコレータは Matrix.__trudiv__ メソッドに付けられているので、A を Matrix クラスのインスタンスにする必要があるかと思います。

    # C = Matrix(2, 2, A)
    # print("C=A/B:")
    A = Matrix(*A.shape, A)
    print("A/B:")
    print(A/B)

それから、すぐに気が付くかと思いますが、変数名を間違えています。

    if np.all(B != 0):
        # return func(a, b)
        return func(A, B)

Comments

[soft_cream]

回答ありがとうございます。
インスタンスについて見直そうと思います
変数についてもありがとうございます。

Answer 2

print(A/B)
はMatrixとndarrayの割り算ではなく、ndarrayとndarrayの割り算です。
print(C/B)
とすると、

>>> print(C/B)
ゼロ除算が発生します

となります。

Comments

[soft_cream]

回答ありがとうございます。
期待した結果が得られました。