# Copyright (c) IMToolkit Development Team
# This toolkit is released under the MIT License, see LICENSE.txt
import unittest
import numpy as np
from imtoolkit.Util import testUnitary
from imtoolkit.DiagonalUnitaryCode import DiagonalUnitaryCode
[docs]class OSTBCodeTest(unittest.TestCase):
[docs] def test_M2(self):
for L in [2, 4, 16, 256]:
testUnitary(2, DiagonalUnitaryCode(2, L))
[docs] def test_M4(self):
for L in [4, 16, 256]:
testUnitary(4, DiagonalUnitaryCode(4, L))
if __name__ == '__main__':
unittest.main()
