def assert_equal(self, tmp, ref):
    assert (np.max(np.abs((tmp - ref))) <= self.tol)
    assert (np.sum(np.abs((tmp - ref))) <= (tmp.shape[0] * self.tol))
def assert_equal(self, tmp, ref):
    assert (np.max(np.abs((tmp - ref))) <= self.tol)
    assert (np.sum(np.abs((tmp - ref))) <= (tmp.shape[0] * self.tol))