import numpy    as np
import unittest as ut

from qfactor.gates import Gate
from qfactor.tensors import CircuitTensor

class TestCircuitTensorConstructor ( ut.TestCase ):

    TOFFOLI = np.asarray(
            [[1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j]] )

    INVALID = np.asarray(
            [[1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j],
            [0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j]] )

    def test_gate_constructor_invalid ( self ):
        gate = Gate( self.TOFFOLI, (0, 1, 2) )
        self.assertRaises( TypeError, CircuitTensor, self.INVALID, [ gate ] )
        self.assertRaises( TypeError, CircuitTensor, self.TOFFOLI, [ "a", gate ] )
        self.assertRaises( TypeError, CircuitTensor, self.TOFFOLI, [ gate, "a" ] )
        self.assertRaises( TypeError, CircuitTensor, self.TOFFOLI, [ "a" ] )
        self.assertRaises( TypeError, CircuitTensor, self.TOFFOLI, "a" )

        gate = Gate( self.TOFFOLI, (0, 1, 7) )
        self.assertRaises( ValueError, CircuitTensor, self.TOFFOLI, [ gate ] )

    def test_gate_constructor_valid ( self ):
        gate = Gate( self.TOFFOLI, (0, 1, 2) )
        ct = CircuitTensor( self.TOFFOLI, [ gate ] )
        self.assertTrue( np.array_equal( gate.utry, ct.gate_list[0].utry ) )
        self.assertTrue( len( ct.gate_list ) == 1 )
        self.assertTrue( np.allclose( ct.utry, np.identity( 8 ) ) )


if __name__ == '__main__':
    ut.main()

