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)
