py_task/venv/lib/python3.9/site-packages/pytils/test/test_utils.py

# -*- coding: utf-8 -*-
"""
Unit-tests for pytils.utils
"""

import decimal
import unittest

import pytils


class ChecksTestCase(unittest.TestCase):
    """
    Test case for check_* utils
    """
        
    def testCheckLength(self):
        """
        Unit-test for pytils.utils.check_length
        """
        self.assertEqual(pytils.utils.check_length("var", 3), None)
        
        self.assertRaises(ValueError, pytils.utils.check_length, "var", 4)
        self.assertRaises(ValueError, pytils.utils.check_length, "var", 2)
        self.assertRaises(ValueError, pytils.utils.check_length, (1,2), 3)

    def testCheckPositive(self):
        """
        Unit-test for pytils.utils.check_positive
        """
        self.assertEqual(pytils.utils.check_positive(0), None)
        self.assertEqual(pytils.utils.check_positive(1), None)
        self.assertEqual(pytils.utils.check_positive(1, False), None)
        self.assertEqual(pytils.utils.check_positive(1, strict=False), None)
        self.assertEqual(pytils.utils.check_positive(1, True), None)
        self.assertEqual(pytils.utils.check_positive(1, strict=True), None)
        self.assertEqual(pytils.utils.check_positive(decimal.Decimal("2.0")), None)
        self.assertEqual(pytils.utils.check_positive(2.0), None)
        
        self.assertRaises(ValueError, pytils.utils.check_positive, -2)
        self.assertRaises(ValueError, pytils.utils.check_positive, -2.0)
        self.assertRaises(ValueError, pytils.utils.check_positive, decimal.Decimal("-2.0"))
        self.assertRaises(ValueError, pytils.utils.check_positive, 0, True)


class SplitValuesTestCase(unittest.TestCase):
    
    def testClassicSplit(self):
        """
        Unit-test for pytils.utils.split_values, classic split
        """
        self.assertEqual(("Раз", "Два", "Три"), pytils.utils.split_values("Раз,Два,Три"))
        self.assertEqual(("Раз", "Два", "Три"), pytils.utils.split_values("Раз, Два,Три"))
        self.assertEqual(("Раз", "Два", "Три"), pytils.utils.split_values(" Раз,   Два, Три  "))
        self.assertEqual(("Раз", "Два", "Три"), pytils.utils.split_values(" Раз, \nДва,\n Три  "))
    
    def testEscapedSplit(self):
        """
        Unit-test for pytils.utils.split_values, split with escaping
        """
        self.assertEqual(("Раз,Два", "Три,Четыре", "Пять,Шесть"), pytils.utils.split_values("Раз\,Два,Три\,Четыре,Пять\,Шесть"))
        self.assertEqual(("Раз, Два", "Три", "Четыре"), pytils.utils.split_values("Раз\, Два, Три, Четыре"))


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