def check_diag(self): a = (100 * get_mat(5)).astype("f") b = a.copy() for k in range(5): for l in range(k + 3, 5): b[k, l] = 0 assert_equal(tril(a, k=2), b) b = a.copy() for k in range(5): for l in range(max((k - 1, 0)), 5): b[k, l] = 0 assert_equal(tril(a, k=-2), b)
def check_basic(self): a = (100 * get_mat(5)).astype("l") b = a.copy() for k in range(5): for l in range(k + 1, 5): b[k, l] = 0 assert_equal(tril(a), b)