QElephant/Matrix_test.py
2023-11-08 14:27:38 +01:00

18 lines
No EOL
457 B
Python

from QElephant import *
import pytest
m = Matrix([[1, 2, 3], [4, 5, 6]])
assert m._Matrix__m == [[1, 2, 3], [4, 5, 6]]
assert m._Matrix__size == (2, 3)
m_ = m * Matrix([[1, 1], [0, 0]])
assert m_._Matrix__m == [
[1, 1, 2, 2, 3, 3],
[0, 0, 0, 0, 0, 0],
[4, 4, 5, 5, 6, 6],
[0, 0, 0, 0, 0, 0],
]
m_ = m._Matrix__apply([1, 2, 3])
assert m_ == [14, 32]
assert str(m) == "[[1, 2, 3], [4, 5, 6]]"
assert m.to_list() == [[1, 2, 3], [4, 5, 6]]