slpkg/tests/test_utils.py

import unittest
from slpkg.utils import Utils


class TestUtils(unittest.TestCase):

    def test_dimensional_list(self):
        """Testing dimesional list util
        """
        lists = [[1, 2, 3, 4, 5]]
        utils = Utils()
        self.assertEqual([1, 2, 3, 4, 5], utils.dimensional_list(lists))

    def test_remove_dbs(self):
        """Testing removing doubles item from list
        """
        lists = [1, 2, 3, 3, 4, 5, 2, 1]
        utils = Utils()
        self.assertEqual([1, 2, 3, 4, 5], utils.remove_dbs(lists))


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