view tests/test_conformity.py @ 187:8aec5ebb2d19

whitespace
author Jeff Hammel <k0scist@gmail.com>
date Fri, 21 Jul 2017 12:50:17 -0700
parents c2f545f32025
children 100697f7c195
line wrap: on
line source

#!/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.NonformantRowLengths as e:
            pass
        assert e is not None
        assert isinstance(e, NonConformantRowLengths)

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