#!/usr/bin/env python

"""
test conformity
"""

import unittest
from numerics import conformity

class TestConformity(unittest.TestCase):
    """tests for ensuring data conformity"""

    def test_equal_lengths(self):

        data = [[1,2,3],
                [4,5,6],
                [7,8,9]]

        assert conformity.ensure_row_length(data) == 3

    def test_nonequal_lengths(self):
        data = [[1,2,3],
                [4,5,6],
                [7,8,9, 10]  # oops!
                ]

        e = None
        try:
            conformity.ensure_row_length(data)
        except conformity.NonConformantRowLengths as e:
            pass
        assert e is not None
        assert isinstance(e, conformity.NonConformantRowLengths)

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