tvii: tests/test_transpose.py comparison

test matrix transposition

comparison

equal deleted inserted replaced

-:5916b04d0582
+:0bb36ae047c3
+#!/usr/bin/env python
+"""
+test matrix data transposition
+"""
+import unittest
+from tvii.transpose import transpose
+from tvii.transpose import NotAMatrix
+class TestTranspose(unittest.TestCase):
+def test_transpose(self):
+data = [[1.0, 0.0],
+[2.0, -1.0],
+[3.0, -2.0],
+[4.0, -3.0]]
+transposed = [[1.0, 2., 3., 4.],
+[0., -1., -2., -3]]
+assert (transpose(data) == transposed)
+def test_not_a_matrix(self):
+not_a_matrix = [[1,2,3],
+[4,5,6],
+[7, 8, 9, 10]  # oops
+]
+e = None
+try:
+transpose(not_a_matrix)
+except NotAMatrix as e:
+pass
+assert e is not None
+assert isinstance(e, NotAMatrix)
+if __name__ == '__main__':
+unittest.main()

Mercurial > hg > tvii