import numpy    as np
import unittest as ut

from qfactor.gates import Gate


class TestGateStr ( 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]] )

    def test_gate_str_1 ( self ):
        gate = Gate( self.TOFFOLI, (0, 1, 2) )
        self.assertEqual( str( gate ),
                          str( (0, 1, 2) ) + ":" + str( self.TOFFOLI ) )

    def test_gate_str_2 ( self ):
        gate = Gate( self.TOFFOLI, (4, 5, 7) )
        self.assertEqual( str( gate ),
                          str( (4, 5, 7) ) + ":" + str( self.TOFFOLI ) )

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

