# HG changeset patch # User Jeff Hammel # Date 1513545376 28800 # Node ID 0bb36ae047c3e1c55f508ce671ab697c188301ef # Parent 5916b04d0582ee4adeeb41e816c06d3bb8ba31fb test matrix transposition diff -r 5916b04d0582 -r 0bb36ae047c3 tests/test_transpose.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/test_transpose.py Sun Dec 17 13:16:16 2017 -0800 @@ -0,0 +1,38 @@ +#!/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()