为包含深度嵌套numpy数组的Python对象实现__eq__

import numpy as np

class CustomObject:
    def __init__(self, nested_array):
        self.nested_array = nested_array

class CustomObject:
    def __init__(self, nested_array):
        self.nested_array = nested_array
    
    def __eq__(self, other):
        if isinstance(other, CustomObject):
            return np.array_equal(self.nested_array, other.nested_array)
        return False

# 创建两个包含深度嵌套numpy数组的对象
array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])
nested_array1 = np.array([array1, array2])
nested_array2 = np.array([array1, array2])

# 创建两个自定义对象
obj1 = CustomObject(nested_array1)
obj2 = CustomObject(nested_array2)

# 比较两个对象是否相等
print(obj1 == obj2)  # 输出：True