view tests/test_unique.py @ 87:9d5a5e9f5c3b

add kmeans + dataset
author Jeff Hammel <k0scist@gmail.com>
date Sun, 17 Dec 2017 14:05:57 -0800
parents 9d55e0299c3f
children
line wrap: on
line source

#!/usr/bin/env python

"""
test uniqueness of set elements
"""

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:
            e = _e
        assert e is not None
        assert isinstance(e, NotUnique)

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