view tests/test_unique.py @ 50:4b20694b8a16

add module + test for uniqueness
author Jeff Hammel <k0scist@gmail.com>
date Sun, 17 Sep 2017 14:28:36 -0700
parents
children e1eb45c4763e
line wrap: on
line source

#!/usr/bin/env python

import os
import unittest
from tvii.unique import unique
from tvii.unique import NotUnique

class TestUnique(unittest.TestCase):

    def test_positive(self):
        """ensure the unique functionality works as expected"""

        # true positive
        a_unique_value = [7] * 7
        assert unique(a_unique_value) == 7

    def test_negative(self):
        """what if a set is not unique?"""

        data = [1,2,3,4]
        e = None
        try:
            unique(data)
        except NotUnique as e:
            pass
        assert e is not None
        assert isinstance(e, NotUnique)

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