permanent/tests/test_permanent.py

import numpy as np
from permanent.permanent import permanent
import pytest


def test_permanent():
    """ Test that basic functions work right """
    m = np.eye(10, dtype=complex)
    assert permanent(m) == 1
    m = np.zeros((10, 10), dtype=complex)
    assert permanent(m) == 0


def test_floaty():
    """ More tests using a precomputed permanent """
    np.random.seed(1234)
    m = np.random.uniform(0, 1, 16) + 1j * np.random.uniform(0, 1, 16)
    m = m.reshape(4, 4)
    p = permanent(m)
    assert np.allclose(p, -8.766131870776363 + 1.072095650303524j)


def test_error():
    """ Should raise a TypeError as we are using the wrong dtype """
    with pytest.raises(TypeError):
        m = np.eye(10, dtype=float)
        permanent(m)