<reponame>wesselb/matrix<gh_stars>1-10
import lab as B

# noinspection PyUnresolvedReferences
from ..util import AssertDenseWarning, approx, check_un_op, dense1, diag1


def test_take_dense(dense1):
    check_un_op(B.isnan, dense1)


def test_take_diag(diag1):
    with AssertDenseWarning('applying "isnan" to <diagonal>'):
        check_un_op(B.isnan, diag1)
