Clean up, remove compat with py < 3.6

[pyi2ncommon] / test / test_type_helpers.py

# The software in this package is distributed under the GNU General
# Public License version 2 (with a special exception described below).
#
# A copy of GNU General Public License (GPL) is included in this distribution,
# in the file COPYING.GPL.
#
# As a special exception, if other files instantiate templates or use macros
# or inline functions from this file, or you compile this file and link it
# with other works to produce a work based on this file, this file
# does not by itself cause the resulting work to be covered
# by the GNU General Public License.
#
# However the source code for this file must still be made available
# in accordance with section (3) of the GNU General Public License.
#
# This exception does not invalidate any other reasons why a work based
# on this file might be covered by the GNU General Public License.
#
# Copyright (c) 2016-2018 Intra2net AG <info@intra2net.com>

""" type_helper_unittest.py: unit tests for type_helpers

Test classes and functions in type_helpers

Should be run from python2 and python3!

For help see :py:mod:`unittest`
"""

import unittest
from sys import version_info

from src.type_helpers import is_unicode, isstr


class TypeHelperTester(unittest.TestCase):

    def test_is_unicode(self):
        """ tests function is_unicode """

        self.assertFalse(is_unicode(1))
        self.assertFalse(is_unicode([]))
        self.assertFalse(is_unicode(unittest.TestCase))
        self.assertFalse(is_unicode(type(unittest.TestCase)))

        self.assertTrue(is_unicode('bla'))
        self.assertTrue(is_unicode(r'bla'))
        self.assertFalse(is_unicode(b'bla'))
        self.assertFalse(is_unicode(br'bla'))

    def test_isstr(self):
        """ test function isstr """

        tests = [
            ('abc', True), (u'abc', True), (r'abc', True),
            (1, False), (['a', 'b', 'c'], False), (('a', 'b', 'c'), False),
            (b'abc', False)]

        for test_var, expected_result in tests:
            self.assertEqual(isstr(test_var), expected_result,
                             'isstr of var {0} (which is of type {1}) returned'
                             '{2} but expected {3}!'
                             .format(test_var, type(test_var), isstr(test_var),
                                     expected_result))


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